anaconda建立虛擬環境安裝pytorch
阿新 • • 發佈:2022-05-18
注意:
- 在安裝之前請選擇儲存空間大於20G
- 儘量不要移動安裝好的anaconda,一些快捷方式會找不到,需要重新配置環境變數
- 安裝完成後我覺得虛擬機器是個好東西,因為我一直在擔心刪除anaconda後再安裝會不會安裝失敗等問題,隨時準備重灌系統
正文:
-
安裝anaconda,使用許可權選all users,不自動新增環境變數,隨後手動新增
-
win+r命令列,或者Anaconda Prompt
-
檢視映象源 conda config --show channels #注意兩個符號
-
(如需要替換)刪除所有映象源 conda config --remove-key channels
-
新增映象源
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 --set show_channel_urls yes 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/peterjc123/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
-
檢視當前已有虛擬環境 conda env list
-
建立虛擬環境 conda create -n env_name python=3.8 #自己python版本,直接在命令列輸入python會顯示版本
-
進入虛擬環境 conda activate env_name
-
安裝pytorch,開啟pytorch官網,向下拖動,找到下載的命令部分,選擇符合電腦版本的選項,將命令複製到命令列執行,可以將命令中 -c刪掉,否則是從預設源下載,速度慢。下載選擇conda和pip應該都可以,本人選擇conda,CUDA只能選擇比自己電腦版本低的下載,在命令列輸入 nvidia-smi 顯示電腦顯示卡資訊,如果沒有nvidia 可以選none或者cpu,版本不夠有兩種選擇:1.升級因特爾顯示卡驅動,比如本人CUDA升級後由11.1變成11.6;2.在pytorch官網下載原來的版本 (這裡下載很慢,檔案比較大)
-
安裝完成,檢驗是否可以使用CUDA加速,返回true,成功
python import torch torch.cuda.is_available()