1.Cuda的下載安裝及配置
首先我們要確定本機(jī)是否有獨(dú)立顯卡。在計(jì)算機(jī)-管理-設(shè)備管理器-顯示適配器中,查看是否有獨(dú)立顯卡。

可以看到本機(jī)有一個(gè)集成顯卡和獨(dú)立顯卡NVIDIA GetForce GTX 1050。
接下來,測試本機(jī)獨(dú)立顯卡是否支持CUDA的安裝,點(diǎn)擊此處查詢顯卡是否在列表中。


從上圖中,可以看到我本機(jī)的獨(dú)立顯卡是支持CUDA安裝的,計(jì)算力是6.1。
那么,接下來就是選擇CUDA進(jìn)行安裝即可,點(diǎn)擊此處


這里我選了CUDA Toolkit10.0的版本,至于選擇哪個(gè)版本,個(gè)人認(rèn)為應(yīng)該沒多大差別,一般就是看這個(gè)版本是否要求GPU的計(jì)算能力是多少以上。大約是2.1G。我用windows的IDM下載工具下載的,速度蠻快的。
接著就是安裝過程,雙擊打開顯示臨時(shí)解壓目錄,不需要改變,默認(rèn)即可。

接下來,進(jìn)入NVIDIA安裝過程,在這安裝過程中,我一開始直接選擇的精簡安裝,但由于VS的原因,導(dǎo)致無法正常安裝,于是我換成了自定義的安裝方式,并將VS勾給去掉,便可以正常安裝了,至于CUDA的安裝目錄,大家默認(rèn)安裝在C盤即可。


安裝完成之后,便是配置環(huán)境變量。環(huán)境變量配置如下圖所示。

測試CUDA是否正常安裝??梢钥吹綔y試成功,接著進(jìn)行cudnn的下載。

2.CUDNN的下載及配置
點(diǎn)擊此處,選擇下載download cudnn,但這里需要你注冊一個(gè)賬號(hào),然后進(jìn)行問卷之后才可以進(jìn)行下載頁面,反正一步步操作即可。
然后因?yàn)槲疑弦徊紺UDA的版本是10.0,而CUDNN的版本要跟CUDA版本一致,所以選擇第二個(gè)下載即可。

下載之后,解壓縮,將CUDNN壓縮包里面的bin、clude、lib文件直接復(fù)制到CUDA的安裝目錄下,直接覆蓋安裝即可。

3.pytorch-gpu的安裝
直接來到Pytorch的官網(wǎng),此處,因?yàn)楦鶕?jù)自己的配置進(jìn)行選擇,我試過用conda安裝,但是conda安裝老是不成功,也替換過清華鏡像源,但速度實(shí)在是龜速,半天沒動(dòng)靜。雖然pip下載也蠻慢,但至少可以安裝成功。這里我對pip的安裝方式,稍作修改,讓其直接從清華鏡像源下載。然后慢慢等待安裝成功即可。
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

然后命令行驗(yàn)證pytorch是否正常安裝成功,這里可以正常打印出版本號(hào),安裝沒問題。
import torch
print(torch.__version__)

但接下來我嘗試讓其輸出是否可以正常調(diào)用CUDA,卻輸出False.
print(torch.cuda.is_available())
經(jīng)過一段時(shí)間的交流,找到問題所在。因?yàn)槲冶緳C(jī)是集成顯卡和獨(dú)立顯卡并存,可能是因?yàn)槟J(rèn)調(diào)用集成顯卡而造成的問題,那么接下來就是將獨(dú)立顯卡設(shè)為默認(rèn)即可。
但我因?yàn)槎啻螄L試的結(jié)果,成功將我本機(jī)的NAVIDIA控制面板給卸載掉了,無法正常打開,所以我還需要重新下載NVIDIA的控制面板,來到官網(wǎng)此處,選擇對應(yīng)的版本下載安裝即可。

然后右鍵打開NIVIDIA控制面板,選擇管理3D設(shè)置,全局設(shè)置中選擇高性能NVIDIA處理器即可。


然后在命令行下重新輸入
import torch
print(torch.cuda.is_available())
即可返回True,若還返回False的話 ,記得重啟一下電腦,遇到問題重啟一下電腦,看是否可以正常顯示。

到此這篇關(guān)于win10下pytorch-gpu安裝以及CUDA詳細(xì)安裝過程的文章就介紹到這了,更多相關(guān)pytorch-gpu安裝及CUDA安裝內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
您可能感興趣的文章:- Linux環(huán)境下GPU版本的pytorch安裝
- PyTorch 檢查GPU版本是否安裝成功的操作
- 詳解anaconda離線安裝pytorchGPU版
- win10使用清華源快速安裝pytorch-GPU版(推薦)
- 解決pytorch-gpu 安裝失敗的記錄