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。

最后安装成功的截图:

win10安装Pytorch【最新版】

测试安装,torch导入没报错,且torch.cuda.is_available()是True就表示支持GPU:

win10安装Pytorch【最新版】

 

转载于:https://www.cnblogs.com/lijinze-tsinghua/p/10973241.html

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/100881.html原文链接:https://javaforall.net

(0)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号