今天折騰了一天，終于是把它折騰出來(lái)了，可能真的是自己太菜了，廢話(huà)不多說(shuō)，簡(jiǎn)單記錄一下。

配置

系統(tǒng) ：win10

python版本 ：3.7.3

顯卡 ：NVIDIA geforce gtx 960M

安裝cuda版本 ：cuda9.0

cudnn版本 ：cudnn7.1

pytorch版本 ：pytorch_1.1.0

編譯工具：pycharm

安裝步驟

1.安裝cuda

（1） 首先檢擦自己電腦顯卡是否支持cuda，這一點(diǎn)網(wǎng)上有很多，沒(méi)啥營(yíng)養(yǎng)，大致就是打開(kāi)設(shè)備管理器，點(diǎn)擊顯示適配器，然后找到自己的顯卡，查看是否安裝NVIDIA顯卡。若安裝，在桌面右鍵打開(kāi)NVIDIA控制面板，點(diǎn)擊右下角的系統(tǒng)信息，點(diǎn)擊組件，在列表中查看自己顯卡可支持的cuda版本，如圖：

也可以對(duì)照這個(gè)表（NVIDIA也給出了對(duì)應(yīng)關(guān)系：傳送門(mén)）查找可以安裝的cuda toolkit的版本（比如我的程序版本號(hào)是385.54，因此選擇cuda9.0）。
注意，這個(gè)驅(qū)動(dòng)是向下兼容的，因此這也決定了其可安裝的最高版本。

（2） 下載cuda
從NVIDIA的官網(wǎng)下載對(duì)應(yīng)版本的cuda toolkit，傳送門(mén)。
并進(jìn)行以下選擇進(jìn)行下載：

（3） 安裝cuda，雙擊.exe安裝包。

會(huì)進(jìn)入安裝界面：

選擇同意并繼續(xù)。
之后彈出安裝選項(xiàng)，選擇自定義模式

并選擇默認(rèn)勾選的驅(qū)動(dòng)程序組件，點(diǎn)擊下一步

選擇默認(rèn)安裝位置進(jìn)行安裝

之后就是漫長(zhǎng)的等待安裝時(shí)間，大部分的人都會(huì)彈出這個(gè)框框，表示安裝成功。

然而，重點(diǎn)來(lái)了，也有不幸運(yùn)的人，比如說(shuō)我，會(huì)顯示安裝失敗，一般會(huì)提示Visual studio integration安裝失敗，其實(shí)原因有很多，需要自己嘗試，列舉相關(guān)的解決辦法：

多嘗試幾次安裝，可能會(huì)有成功的可能性；

在自定義階段選擇驅(qū)動(dòng)程序組件時(shí)，將CUDA中Visual studio integration前面的對(duì)勾去掉進(jìn)行安裝，安裝過(guò)后嘗試單獨(dú)對(duì)CUDA中Visual studio integration進(jìn)行安裝或者直接整體再重裝，多嘗試兩次，網(wǎng)上有很多人都是試了好幾次成功的；

如果上邊的方式?jīng)]能解決你的問(wèn)題（跟我一樣的情況），嘗試使用我使用的方法：因?yàn)槊看蜗到y(tǒng)更新可能會(huì)跟著更新NVIDIA顯卡的驅(qū)動(dòng)，這可能會(huì)導(dǎo)致我們?cè)诎惭bcuda時(shí)其不能正常安裝，因此需要從新安裝NVIDIA的顯卡驅(qū)動(dòng)，注意：敲黑板，千萬(wàn)別用其它第三方軟件進(jìn)行安裝，可能裝完還是不能用，從官網(wǎng)上下載或者直接刪掉原來(lái)的文件，破壞現(xiàn)在裝好的驅(qū)動(dòng)，因?yàn)橄螺d的CUDA安裝包中包含顯卡驅(qū)動(dòng)，所以我是進(jìn)行刪除的，操作步驟：

打開(kāi)計(jì)算機(jī)中的“服務(wù)”，停止所有與nvidia相關(guān)的服務(wù)（應(yīng)該是有四五個(gè)，我從服務(wù)里關(guān)了四個(gè)，從“詳細(xì)信息”里關(guān)了一個(gè)）；

刪除C:\ProgramData, C:\Program Files, C:\Program Files(x86)中和NVIDIA相關(guān)所有文件夾（必須刪除干凈）；

打開(kāi)CUDA進(jìn)行安裝（顯卡驅(qū)動(dòng)會(huì)同時(shí)安裝，不需要另外安裝），至此問(wèn)題解決。
應(yīng)該會(huì)成功安裝，檢查一下：

如果還是不能，就只能在試試其它辦法了。

2.安裝cudnn

cuda安裝完成之后，還需要下載與CUDA對(duì)應(yīng)的相應(yīng)版本的cuDNN，到下圖所示的下載頁(yè)面，下載完成后，將這個(gè)壓縮包里的所有文件放到CUDA10安裝目錄相應(yīng)文件夾下即可。步驟：
下載地址：傳送門(mén)
注意：以上鏈接，初次登陸需要注冊(cè)賬號(hào)。

下載之后，

(1)解壓：會(huì)生成cuda/include、cuda/lib、cuda/bin三個(gè)目錄；

(2)分別將cuda/include、cuda/lib、cuda/bin三個(gè)文件替換掉C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9對(duì)應(yīng)的include、lib、bin文件即可。

添加環(huán)境變量：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9\lbin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9\linclude
此電腦→“高級(jí)系統(tǒng)設(shè)置”→“環(huán)境變量”→“系統(tǒng)變量”→“path”→“編輯”→“新建”加入該路徑即可。

至此cuda和cudnn安裝完畢。

3.安裝pytorch-gpu

（1） 首先創(chuàng)建虛擬環(huán)境：
win+R 輸入cmd，輸入：

conda create -n pytorch-gpu python==3.7.3

完成名為pytorch-gpu的虛擬環(huán)境創(chuàng)建，可以輸入以下命令查看計(jì)算機(jī)中虛擬環(huán)境的情況：

conda env list

而后激活虛擬環(huán)境，輸入以下命令：

conda activate pytorch-gpu

（2） 安裝gpu版本pytorch
激活虛擬環(huán)境后，在當(dāng)前環(huán)境下安裝gpu版本pytorch。
首先要查看cuda9.0適合的pytorch的版本，傳送門(mén)。
點(diǎn)進(jìn)鏈接之后，查找CUDA 9.0 可以安裝的最高版本的pytorch和torchvision的安裝口令，比如我找的是：

# CUDA 10.0 conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch

在pytorch-gpu虛擬環(huán)境下輸入以上口令進(jìn)行pytorch和torchvision的安裝，這個(gè)口令是直接從官網(wǎng)下載，如果網(wǎng)絡(luò)允許應(yīng)該能夠安裝成功。

但是，我發(fā)現(xiàn)我的安裝的很慢（有人說(shuō)提前設(shè)置好鏡像地址，而后將-c pytorch去掉會(huì)安裝的很快，但是我沒(méi)有成功），由于過(guò)程中會(huì)安裝一些其他的必須的工具包，一些小的工具包還能正常安裝，但是一些稍微大一點(diǎn)的安裝包就無(wú)法安裝，其中主要是pytorch和torchvision兩個(gè)包。

于是我想嘗試離線(xiàn)安裝，做法是：當(dāng)碰到這兩個(gè)包在安裝時(shí)直接ctrl+c強(qiáng)制終止，而后在安裝信息中找到對(duì)應(yīng)的文件名稱(chēng)，從這個(gè)國(guó)內(nèi)鏡像網(wǎng)站中找到對(duì)應(yīng)的文件進(jìn)行下載。我下載的是：(如果是其它工具包安裝緩慢，也可以使用此方法進(jìn)行安裝)
pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2
torchvision-0.3.0-py37_cu90_1.tar.bz2

而后在pytorch-gpu虛擬環(huán)境下cd到這個(gè)文件的保存位置進(jìn)行安裝，輸入：

conda install --offline pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2 conda install --offlin torchvision-0.3.0-py37_cu90_1.tar.bz2

從而完成對(duì)pytorch和torchvision的離線(xiàn)安裝。
還沒(méi)有結(jié)束，離線(xiàn)安裝好后，還是需要重新利用conda來(lái)在線(xiàn)安裝一次，要用它來(lái)對(duì)相關(guān)的包來(lái)進(jìn)行在線(xiàn)匹配(也就是根據(jù)安裝好的pytorch來(lái)匹配那些對(duì)應(yīng)的好多庫(kù)，不是說(shuō)再下載一遍pytorch)。
但是一定要注意了!!!
在利用官網(wǎng)代碼時(shí)conda install pytorch torchvision cudatoolkit=9.0 -c pytorch一定要刪掉后面的-c pytorch，因?yàn)樗碇苯訌墓倬W(wǎng)去下載，不會(huì)與本地安裝好的匹配，還會(huì)全部重新下載。所以改成：conda install pytorch torchvision cudatoolkit=9.0它就會(huì)自動(dòng)匹配之前離線(xiàn)安裝好的東西，然后匹配相關(guān)的庫(kù)的等級(jí)、版本之類(lèi)的。

至此安裝完畢，檢驗(yàn)是否安裝成功：
在pytorch-gpu虛擬環(huán)境中輸入：

>>python >>import torch >>print(torch.cuda.is_available())

效果圖如下：

輸出為Ture，表示安裝成功。

4.配置虛擬環(huán)境

使用的工具是pycharm，在pycharm中使用conda虛擬環(huán)境pytorch-gpu。
將自己創(chuàng)建的conda虛擬環(huán)境pytorch-gpu添加到pycharm的interpreter可選列表中：

在pycharm新建工程，或在files->settings->project:->project interpreter界面，可以設(shè)定project interpreter;

新建工程界面，選擇Existing enviroment，點(diǎn)擊…按鈕，會(huì)彈出添加界面;

settings界面，點(diǎn)擊齒輪按鈕，選擇Add，會(huì)彈出添加界面;

在添加界面左側(cè)選擇Conda Environment，右邊選擇Existing environment，再點(diǎn)擊…按鈕添加一個(gè)已經(jīng)存在的conda虛擬環(huán)境，即剛剛構(gòu)建的pytorch-gpu。

使用已經(jīng)添加進(jìn)來(lái)的虛擬環(huán)境pytorch-gpu，在interpreter列表中選擇剛剛添加的nterpreter。

至此全部完成。

寫(xiě)的有點(diǎn)亂，能力有限，記錄一下，希望能夠幫到你。

總結(jié)

以上是生活随笔為你收集整理的win10+NVIDIA geforce gtx 960M+cuda9.0+cudnn7.1+pytorch_1.1.0+torchvision_0.3.0---已解决的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。