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:

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM