在網(wǎng)速不好的情況下，如何用離線的方式安裝pytorch。這里默認(rèn)大家已經(jīng)安裝了anaconda了。

安裝Nvidia驅(qū)動、cuda、cudnn等依賴

首先安裝vs社區(qū)版，如果已經(jīng)安裝過可以跳過這一步，下載地址
安裝以下兩個組件即可，不用全部裝上。

之后安裝nvidia驅(qū)動，注意自己顯卡和驅(qū)動的對應(yīng)關(guān)系，下載地址
我的顯卡是940M，對應(yīng)如下選項：

安裝cuda

這里要注意查看驅(qū)動和cuda的對應(yīng)關(guān)系，首先查看自己下載的驅(qū)動文件名，

可以看到最開始有個數(shù)字，這個就是驅(qū)動版本，和cuda會有下圖類似的關(guān)系，由于cuda和驅(qū)動會不斷更新，后面安裝的同學(xué)，可以去google一下，《cuda和nvidia驅(qū)動的對應(yīng)關(guān)系》，獲取最新的信息。

這里我下載的是10.2版本的cuda

cudnn的安裝
安裝cudnn要注冊一個nvidia賬號，才可以進行下載，這個也要注意版本的對應(yīng)關(guān)系。

確定pytorch安裝所需要的包

首先我們創(chuàng)建一個新的conda 環(huán)境，在這個環(huán)境中我們安裝pytorch-gpu版本。

conda create -n pytorch pip python=3.8

pytorch是我們建立的環(huán)境名，3.8是該環(huán)境的python版本，都可以自己換。
下一步試著安裝pytorch-gpu包。
首先訪問pytorch的官網(wǎng)
選擇對應(yīng)選項后會生成對應(yīng)的命令

我們在anaconda命令行中輸入

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

會發(fā)現(xiàn)有類似的列表

出現(xiàn)選項，我們選no，之后把其中比較大的文件下載到本地。

這是我安裝是的對應(yīng)鏈接，如果版本升級，跳到該文件對應(yīng)的上一級目錄搜索文件名就行了。
接下來安裝這些離線包

conda install --offline 包名

這里注意一點，從安裝包路徑直接運行conda命令需要提前把conda的Scripts目錄加入到環(huán)境變量中。
做好這些準(zhǔn)備之后，就可以再次運行之前的命令安裝pytorch-gpu了。

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

到此這篇關(guān)于詳解anaconda離線安裝pytorchGPU版的文章就介紹到這了,更多相關(guān)anaconda離線安裝pytorchGPU內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論