1. 程式人生 > 程式設計 >詳解anaconda離線安裝pytorchGPU版

詳解anaconda離線安裝pytorchGPU版

在網速不好的情況下,如何用離線的方式安裝pytorch。這裡預設大家已經安裝了anaconda了。

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

首先安裝vs社群版,如果已經安裝過可以跳過這一步,下載地址
安裝以下兩個元件即可,不用全部裝上。

詳解anaconda離線安裝pytorchGPU版

詳解anaconda離線安裝pytorchGPU版

之後安裝nvidia驅動,注意自己顯示卡和驅動的對應關係,下載地址
我的顯示卡是940M,對應如下選項:

詳解anaconda離線安裝pytorchGPU版

安裝cuda

這裡要注意檢視驅動和cuda的對應關係,首先檢視自己下載的驅動檔名,

詳解anaconda離線安裝pytorchGPU版

可以看到最開始有個數字,這個就是驅動版本,和cuda會有下圖類似的關係,由於cuda和驅動會不斷更新,後面安裝的同學,可以去google一下,《cuda和nvidia驅動的對應關係》,獲取最新的資訊。

詳解anaconda離線安裝pytorchGPU版

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

詳解anaconda離線安裝pytorchGPU版

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

確定pytorch安裝所需要的包

首先我們建立一個新的conda 環境,在這個環境中我們安裝pytorch-gpu版本。

conda create -n pytorch pip python=3.8

pytorch是我們建立的環境名,3.8是該環境的python版本,都可以自己換。
下一步試著安裝pytorch-gpu包。
首先訪問pytorch的官網
選擇對應選項後會生成對應的命令

詳解anaconda離線安裝pytorchGPU版

我們在anaconda命令列中輸入

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

會發現有類似的列表

詳解anaconda離線安裝pytorchGPU版

出現選項,我們選no,之後把其中比較大的檔案下載到本地。

包名 清華映象地址
mkl2020 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/mkl-2020.2-256.conda
pytorch-gpu https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.6.0-py3.8_cuda102_cudnn7_0.tar.bz2
cuda-toolkit https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/cudatoolkit-10.2.89-hfd86e86_1.conda

這是我安裝是的對應連結,如果版本升級,跳到該檔案對應的上一級目錄搜尋檔名就行了。
接下來安裝這些離線包

conda install --offline 包名

這裡注意一點,從安裝包路徑直接執行conda命令需要提前把conda的Scripts目錄加入到環境變數中。
做好這些準備之後,就可以再次執行之前的命令安裝pytorch-gpu了。

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

到此這篇關於詳解anaconda離線安裝pytorchGPU版的文章就介紹到這了,更多相關anaconda離線安裝pytorchGPU內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!