清華源安裝GPU版—Pytorch 1.2.0

  • 用conda安裝Pytorch過程中會連接失敗,這是因爲Anaconda.org的服務器在國外,需要切換到國內鏡像源:
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 --set show_channel_urls yes
  • 創建虛擬環境並激活之,在該環境下安裝下面的包:
 conda create  pytorch  python=3.6
  • 安裝pytorch
    conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0

     

發佈了177 篇原創文章 · 獲贊 18 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章