Win10筆記本CUDA+cuDNN+Anaconda+Pycharm配置記錄

注1:這是給我的windows配置的記錄

注2:已有安裝包/安裝程序均在D:\installpackage

注3:今年上半年寫的了,懶得排版就發(fā)發(fā)好了

CUDA + cuDNN

關(guān)系:顯卡,顯卡驅(qū)動(dòng),nvcc,cuda driver,cuda toolkit,cuDNN

0.?如果沒(méi)有nvidia driver,也要安裝!

1. 首先安裝CUDA(已有),要注意CUDA版本和顯卡驅(qū)動(dòng)版本是一一對(duì)應(yīng)的

2.?配置環(huán)境變量

3. 測(cè)試安裝成功否(打開(kāi)命令行窗口,輸入nvcc -V)

4. 再用官方示例測(cè)試,切換到CUDA自帶測(cè)試用例路徑:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite。打開(kāi)cmd,運(yùn)行.\deviceQuery.exe得到下面結(jié)果說(shuō)明安裝成功。 注:這一步可能會(huì)失敗,但是應(yīng)該不太影響后續(xù)(還不清楚原因)。

5. 登錄NVIDIA并下載深度學(xué)習(xí)GPU加速庫(kù)cuDNN(已有)

6. 將解壓后得到的.h .lib 和.dll 文件分別復(fù)制到cuda的include, lib/x64, bin 文件夾下(后者路徑C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2)

7. 用官方示例測(cè)試,運(yùn)行CUDA的官方示例,在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite路徑下打開(kāi)cmd,運(yùn)行.\deviceQuery.exe顯示如下結(jié)果說(shuō)明安裝成功。

查詢

cmd輸入以下內(nèi)容,查看顯卡驅(qū)動(dòng)版本(Driver Version: 442.19),顯卡版本(GeForce MX250)

nvidia-smi

cmd輸入以下內(nèi)容,查詢CUDA版本(V10.2.89)

nvcc --version

Anaconda + Pycharm

1. 安裝Anaconda(已有),新建環(huán)境

2. 下載安裝Pycharm(社區(qū)版即可,已有)

3.?使用自己創(chuàng)建的環(huán)境配置Pycharm

4.?更換清華鏡像源

5. 查看需要下載的pytorch版本,用conda下載,去掉“-c python"便不再?gòu)?qiáng)制從官網(wǎng)下載了,參考 (torchvision,cudatoolkit會(huì)一并下載)

6. conda下載cudnn(如果已有這個(gè)包,就不必了) 注意:進(jìn)入自己的環(huán)境再用此命令,否則會(huì)裝到base下

conda install cudnn

7. 測(cè)試是否能使用CUDA

importtorch

flag=torch.cuda.is_available()

print(flag)

?

ngpu=1

# Decide which device we want to run on

device=torch.device("cuda:0"if(torch.cuda.is_available()andngpu>0)else"cpu")

print(device)

print(torch.cuda.get_device_name(0))

print(torch.rand(3,3).cuda())

如果flag=false

* 如果出現(xiàn)flag是False,檢查一下驅(qū)動(dòng)版本,有時(shí)需要手動(dòng)下載最新的驅(qū)動(dòng)(高級(jí)設(shè)置-清潔安裝)。

* 檢查pytorch版本,可能是pytorch與CUDA版本沖突

同等環(huán)境下使用python3.7,pytorch1.3,cudatoolkit10.1,上述示例一直失敗,但是tensorflow可以正常使用GPU。挨個(gè)排查,直到修改完pytorch版本為1.2才成功。

配置環(huán)境

OS:win10(×64)

顯卡:AMD GeForce MX250

顯卡驅(qū)動(dòng):Driver Version: 442.19

CUDA:V10.2.89

cuDNN:7.6.5(與CUDA版本對(duì)應(yīng))

torch36環(huán)境:python3.6、pytorch(1.2.0,py3.6_cuda100_cudnn7_1)、cudatoolkit(10.0.130)、cudnn(7.6.5)

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容