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

安裝Nvidia驅(qū)動(dòng)、cuda、cudnn等依賴(lài)

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

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

安裝cuda

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

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

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

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

確定pytorch安裝所需要的包

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

conda create -n pytorch pip python=3.8

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

我們?cè)赼naconda命令行中輸入

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

會(huì)發(fā)現(xiàn)有類(lèi)似的列表

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

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

conda install --offline 包名

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

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

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

最新評(píng)論