win10安裝Pytorch【最新版】

因爲2019年4月16日和25日清華和中科大分別宣佈中止Anaconda鏡像服務,所以從2019年5月開始安裝Pytorch都會出現一下錯誤:json

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。url

最後安裝成功的截圖:spa

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

相關文章
相關標籤/搜索