在網速不好的情況下，如何用離線的方式安裝pytorch。這里默認大家已經安裝了anaconda了。

安裝Nvidia驅動、cuda、cudnn等依賴

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

之后安裝nvidia驅動，注意自己顯卡和驅動的對應關系，下載地址
我的顯卡是940M，對應如下選項：

安裝cuda

這里要注意查看驅動和cuda的對應關系，首先查看自己下載的驅動文件名，

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

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

cudnn的安裝
安裝cudnn要注冊一個nvidia賬號，才可以進行下載，這個也要注意版本的對應關系。

確定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的官網
選擇對應選項后會生成對應的命令

我們在anaconda命令行中輸入

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

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

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

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

conda install --offline 包名

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

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

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

最新評論