1. 程式人生 > 實用技巧 >win10安裝Pytorch【最新版】

win10安裝Pytorch【最新版】

win10安裝Pytorch【最新版】

由於2019年4月16日和25日清華和中科大分別宣佈停止Anaconda映象服務,因此從2019年5月開始安裝Pytorch都會出現一下錯誤:

CondaHTTPError: HTTP 404 NOT FOUND for url <https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/noarch/repodata.json

由於本人之前的conda源為清華源,因此這裡需要使用以下命令先刪除清華的源,之後換成上交的源:

conda config --remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/cloud/conda-forge/

之後執行安裝Pytorch的命令即可,本人電腦為win10+Python3.6+CUDA9.1.85:

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

安裝torchvision的目的為torchvision中主要集成了一些資料集、深度學習模型和一些轉換等,torchvision包由三個子包組成,分別是:torchvision.datasets,torchvision.models,torchvision.transforms。

最後安裝成功的截圖:

測試安裝,torch匯入沒報錯,且torch.cuda.is_available()是True就表示支援GPU:

posted @ 2019-06-04 14:08 予你心安 閱讀(...) 評論(...) 編輯 收藏