1.去清華鏡像源下載新版anaconda
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
下載后放入服務器的自己data檔案夾下
2.安裝
bash Anaconda3-xxxxx-Linux-x86_64.sh
一直按回車,直到出現:
輸入yes
這一步為安裝目錄,建議安裝在data下如圖:
回車等待安裝,最后一步會彈出這個框,
輸入yes即可,
3.添加清華鏡像源
進入自己的home檔案夾,找到.condarc檔案,打開,輸入以下代碼
channels:
- defaults
show_channel_urls: true
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
custom_channels:
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: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
保存后關閉即可,
若無.condarc檔案,在終端輸入vim ~/.condarc,接著輸入上面的代碼段保存退出即可,附上vim保存退出教程https://jingyan.baidu.com/article/6fb756ec72b031241858fb90.html
接著在終端依次輸入:
conda clean -i
conda config --set show_channel_urls yes
conda config --set channel_priority flexible
如圖所示:
4.創建虛擬環境
首先輸入
nvidia-smi
查看cuda最高版本,
再根據下表選擇對應的python版本,
因為cuda版本為10.1,這里的選擇是3.8版本的python(.8服務器cuda為10.0)
輸入如下代碼,即可創建虛擬環境
conda create -n py python=3.8
出現上圖后 按y即可,
出現該指令說明成功安裝,
5.進入虛擬環境:
輸入(每次訓練都要輸入該指令),會發現我們的環境變成py
conda activate py
6.安裝torch
進入pytorch官網,尋找對應版本的指令,
以我為例:我是10.1的CUDA與3.8的python,在根據第四步中的對應圖,我選擇的是
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
輸入后就可以成功安裝啦,
附錄:
因為我也是新手,安裝程序中踩了不少坑,特此留下筆記,感謝林師兄的幫助,本文基于林師兄的CSDN進行補充,
參考:
https://blog.csdn.net/weixin_49643423/article/details/108896440
pytorch 不同版本對應的cuda - cltt - 博客園
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/439203.html
標籤:AI