如何使用Conda源快速安裝PyTorch?


數據科學交流群,群號:189158789 ,歡迎各位對數據科學感興趣的小伙伴的加入!

1.切換國內源

 

 

1.1.在家目錄生成.condarc

conda config --set show_channel_urls yes

1.2.修改.condarc文件

show_channel_urls: true
channels:
  - defaults
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/simpleitk

1.3.清除索引緩存,保證用的是鏡像站提供的索引

conda clean -i

2.安裝mamba

 

 

Conda作為使用最為廣泛的數據科學環境管理工具,可以協助我們很方便的完成創建管理環境、下載安裝第三方庫、軟件包等操作,但其在下載資源的過程中下載速度時常令人捉急,即使使用連接速度更快的國內鏡像,也擺脫不了其單線程挨個下載資源導致的低效問題。

Mamba(黑曼巴)專為加速Conda而生,其改寫了Conda下載資源的固有方式,以多線程的方式對網絡資源進行並行下載,從而大幅提升Conda效率,以下是安裝mamba的方法:

conda install mamba

3.在線安裝PyTorch

 

 

3.1.到官網獲取安裝方式:

https://pytorch.org/get-started/locally/

可以看到里面的命令是這個:

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

3.2.使用mamba安裝PyTorch

注意這里我們呢要去掉-c,這樣才會從默認的清華源來進行安裝,並且我們要使用mamba來加速

mamba install pytorch torchvision torchaudio cudatoolkit=10.2

4.離線安裝PyTorch

 

 

如果你覺得mamba的安裝還是比較慢,可以使用專業的下載工具,比如迅雷。

但是在下載之前你要明白你要知道安裝包的下載地址:

 

 如上圖所示,包在解析的時候一般會告訴你地址,用瀏覽器直接訪問這些地址,使用Ctrl+F來搜索具體的包名即可找到。

下載好了以后我們可以使用

conda install --use-local ……

把……替換成你的文件的絕對路徑名,如果是windows10,你可以直接按住shift+鼠標右鍵,然后點擊“復制為路徑”即可

此時粘貼到命令行終端即可。例如:

conda install --user-local "F:\迅雷雲盤\pytorch-1.7.1-py3.8_cuda102_cudnn7_0.tar.bz2"

 

注意:由於反復多次下載,很有可能會產生上次的下載內容會對下一次安裝產生問題,此時,可以看看報錯信息,找到對應的路徑刪除同名的文件夾即可,然后再用conda install --use-local 來安裝就不會再報錯了!!!

5.驗證一下安裝:

 

 

進入python環境中:

python

運行:

import torch
x = torch.rand(5, 3)
print(x)

檢查Cuda:

import torch
torch.cuda.is_available()

 

最后,祝你快快安裝成功,以便能夠進入深度學習的學習階段!!!

 

數據科學交流群,群號:189158789 ,歡迎各位對數據科學感興趣的小伙伴的加入!


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM