首頁 > 軟體

教你兩步解決conda安裝pytorch時下載速度慢or超時的問題

2023-03-31 06:01:43

1.為conda設定清華源

開啟cmd輸入以下命令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --set show_channel_urls yes

2.建立新的虛擬環境(若已建立,請忽略)

如果你是第一次安裝Miniconda,那麼只存在base環境。由於還要安裝Pytorch,所以要再建立一個新的虛擬環境。我們可以利用如下命令建立環境:

conda create -n xxxx(環境名) python=x.x(python版本) 

例如:

conda create -n pytorch python=3.6

建立完成後,在命令列輸入conda env list就能看到目前已經建立的環境(我的環境名為gluon,所以下圖中可以看到除了base外還有個gluon):

3.啟用虛擬環境

在命令列輸入conda activate 環境名即可啟用環境,如下圖所示:

4.安裝pytorch

在pytorch官網中選擇你要安裝的pytorch版本,下方就會給出相應的安裝命令。比如這裡我用的是win10+cuda11.3並且選擇用pip安裝:

下面是重點:
在安裝命令後加上–default-timeout=1000,修改下載超時判定的條件為1s(這麼幹不僅降低了超時的概率,似乎還提高了下載速度(?玄學))
比如我用的是:

pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio===0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html --default-timeout=1000

輸入命令後敲個回車就等它自己慢慢下載了…

最後在Python中import一下然後列印一下torch版本號,驗證是否安裝成功:

或者在虛擬環境中使用pip list顯示所有已安裝的包,可以找到torch及torchvision:

到此這篇關於教你兩步解決conda安裝pytorch時下載速度慢or超時的問題的文章就介紹到這了,更多相關解決安裝pytorch下載慢內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com