解決OSError: CUDA_HOME environment variable is not set. Please set it to your CU

在windows。anaconda虛擬環(huán)境下安裝pytorch的C++Extension的時(shí)候出現(xiàn)

創(chuàng)新互聯(lián)公司秉承實(shí)現(xiàn)全網(wǎng)價(jià)值營(yíng)銷(xiāo)的理念，以專業(yè)定制企業(yè)官網(wǎng)，成都網(wǎng)站設(shè)計(jì)、網(wǎng)站制作，微信小程序開(kāi)發(fā)，網(wǎng)頁(yè)設(shè)計(jì)制作，成都做手機(jī)網(wǎng)站，成都營(yíng)銷(xiāo)網(wǎng)站建設(shè)幫助傳統(tǒng)企業(yè)實(shí)現(xiàn)“互聯(lián)網(wǎng)+”轉(zhuǎn)型升級(jí)專業(yè)定制企業(yè)官網(wǎng),公司注重人才、技術(shù)和管理，匯聚了一批優(yōu)秀的互聯(lián)網(wǎng)技術(shù)人才,對(duì)客戶都以感恩的心態(tài)奉獻(xiàn)自己的專業(yè)和所長(zhǎng)。原因

C++Extension有對(duì)CUDA的依賴，并且此cuda需要是電腦安裝的而不是使用anaconda下載的cudatookit。具體原因請(qǐng)見(jiàn)：cuda和cudatoolkit_獨(dú)孤的大山貓的博客-博客_cudatoolkit和cuda有關(guān)系嗎

在大多數(shù)情況下，上述 cudatoolkit 是可以滿足 Pytorch 等框架的使用需求的。但對(duì)于一些特殊需求，如需要為 Pytorch 框架添加 CUDA 相關(guān)的拓展時(shí)( Custom C++ and CUDA Extensions )，需要對(duì)編寫(xiě)的 CUDA 相關(guān)的程序進(jìn)行編譯等操作，則需安裝完整的 Nvidia 官方提供的 CUDA Toolkit.

解決：

下載CUDA。CUDA Toolkit - Free Tools and Training | NVIDIA Developer
安裝后在系統(tǒng)環(huán)境變量設(shè)置可以看到（路徑為默認(rèn)安裝路徑）

再添加一個(gè)與CUDA_PATH相同路徑的CUDA_HOME

重啟計(jì)算機(jī)，即可解決

你是否還在尋找穩(wěn)定的海外服務(wù)器提供商？創(chuàng)新互聯(lián)www.cdcxhl.cn海外機(jī)房具備T級(jí)流量清洗系統(tǒng)配攻擊溯源，準(zhǔn)確流量調(diào)度確保服務(wù)器高可用性，企業(yè)級(jí)服務(wù)器適合批量采購(gòu)，新人活動(dòng)首月15元起，快前往官網(wǎng)查看詳情吧

網(wǎng)頁(yè)題目：解決OSError:CUDA-創(chuàng)新互聯(lián)
文章轉(zhuǎn)載：http://weahome.cn/article/gopcd.html