先去官網看一下情況https://pytorch.org/get-started/locally/
按順序選擇pytorch版本,選擇linux系統,選擇conda,選擇python版本,選擇cuda版本,然后會給你對應的安裝指令。
接下來就是conda換源,2019年初有段日子清華被迫關閉了conda源的服務,但是7月的時候又拿到了conda的授權,因此清華的conda鏡像服務又開啟了。
清華的conda鏡像使用文檔https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
先運行conda config,這會在你用戶家目錄下添加一個空的.condarc文件,進去把花括號刪了,換成清華文檔里提供的配置代碼。
現在萬事大吉了,運行pytorch官網給你的安裝指令即可。