1. 程式人生 > 其它 >anaconda建立虛擬環境安裝pytorch

anaconda建立虛擬環境安裝pytorch

注意:

  1. 在安裝之前請選擇儲存空間大於20G
  2. 儘量不要移動安裝好的anaconda,一些快捷方式會找不到,需要重新配置環境變數
  3. 安裝完成後我覺得虛擬機器是個好東西,因為我一直在擔心刪除anaconda後再安裝會不會安裝失敗等問題,隨時準備重灌系統

正文:

  1. 安裝anaconda,使用許可權選all users,不自動新增環境變數,隨後手動新增

  2. win+r命令列,或者Anaconda Prompt

  3. 檢視映象源 conda config --show channels #注意兩個符號

  4. (如需要替換)刪除所有映象源 conda config --remove-key channels

  5. 新增映象源

    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/
    
  6. 檢視當前已有虛擬環境 conda env list

  7. 建立虛擬環境 conda create -n env_name python=3.8 #自己python版本,直接在命令列輸入python會顯示版本

  8. 進入虛擬環境 conda activate env_name

  9. 安裝pytorch,開啟pytorch官網,向下拖動,找到下載的命令部分,選擇符合電腦版本的選項,將命令複製到命令列執行,可以將命令中 -c刪掉,否則是從預設源下載,速度慢。下載選擇conda和pip應該都可以,本人選擇conda,CUDA只能選擇比自己電腦版本低的下載,在命令列輸入 nvidia-smi 顯示電腦顯示卡資訊,如果沒有nvidia 可以選none或者cpu,版本不夠有兩種選擇:1.升級因特爾顯示卡驅動,比如本人CUDA升級後由11.1變成11.6;2.在pytorch官網下載原來的版本 (這裡下載很慢,檔案比較大)

  10. 安裝完成,檢驗是否可以使用CUDA加速,返回true,成功

    python         
    import torch 
    torch.cuda.is_available()