Windows10+anacond+GPU+pytorch安裝詳細(xì)過(guò)程

1、查看自己電腦是否匹配GPU版本。

設(shè)備管理器查看。

查看官網(wǎng)是否匹配。地址：https://developer.nvidia.com/cuda-gpus  **

2、進(jìn)入NVIDIA對(duì)電腦版本進(jìn)行查**看。

如果可以的的話(huà)可以自己卸載原來(lái)版本，后安裝新版本。安裝地址https://developer.nvidia.com/cuda-toolkit-archive

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

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

檢驗(yàn)是否安裝成功。

3.CUDNN安裝

點(diǎn)擊https://developer.nvidia.com/rdp/cudnn-download ，選擇下載download cudnn，但這里需要你注冊(cè)一個(gè)賬號(hào)，然后進(jìn)行問(wèn)卷之后才可以進(jìn)行下載頁(yè)面，反正一步步操作即可。

這里記住一定要相配合，不然后期安裝會(huì)失敗。下載之后，解壓縮，將CUDNN壓縮包里面的bin、clude、lib文件直接復(fù)制到CUDA的安裝目錄下，直接覆蓋安裝即可。

到此為止cuda、cudnn安裝成功

**4、安裝anacond

一直點(diǎn)next

第一個(gè)不勾選就可以了。具體過(guò)程可以看https://jingyan.baidu.com/article/425e69e615da75be14fc1645.html

5、安裝pytorch

如果安裝tensorflow，前面步驟一模一樣。只是安裝口令不同罷了。這里只詳細(xì)介紹pytorch的安裝過(guò)程。

打開(kāi)anaconda下的Anaconda Prompt粘貼命令就可以安裝了，此過(guò)程看網(wǎng)速，一般來(lái)說(shuō)很快。但是一般我喜歡激活一個(gè)環(huán)境，首先采用：conda create --name pytorch python=3.6取一個(gè)虛擬環(huán)境。其次activate pytorch激活，在環(huán)境里面安裝，這樣即使安裝tensorflow再安裝一個(gè)環(huán)境就行了。方便。

下面是安裝命令。如果按裝較慢，可以試試清華鏡像來(lái)進(jìn)行安裝。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123conda config --set show_channel_urls yes

最后輸入官網(wǎng)復(fù)制的命令：conda install pytorch torchvision cudatoolkit=10.0即可完成安裝（原命令conda install pytorch torchvision cudatoolkit=10.0 -c pytorch需要去掉后面部分不然還是會(huì)默認(rèn)下載，則鏡像源沒(méi)用）

如果還是安裝不了，建議訪(fǎng)問(wèn)這個(gè)博主的文章https://blog.csdn.net/qq_38704904/article/details/95192856 看一下在本地下載好pytorch后的安裝過(guò)程，其實(shí)也比較簡(jiǎn)單。

另外附上pytorch各個(gè)版本下載地址大全：https://blog.csdn.net/qq_27825451/article/details/90033909

另外這個(gè)地方請(qǐng)注意，安裝torchvision要與torch相匹配。

不然會(huì)出現(xiàn)下面這張錯(cuò)誤

6、測(cè)試是否成功

總結(jié)

到此這篇關(guān)于Windows10+anacond+GPU+pytorch安裝詳細(xì)過(guò)程的文章就介紹到這了,更多相關(guān)Windows10 anacond GPU pytorch安裝內(nèi)容請(qǐng)搜索億速云以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持億速云！