贴上我看的教程https://zhuanlan.zhihu.com/p/26871672

一开始想用pycharm装pytorch,但不知道为什么一直失败。后来只能conda pip安装

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。

但conda pip安装太慢,所以找了镜像。非常快

conda install pytorch=0.4.0 -c soumith

安装成功后测试cuda可不可用,torch.cuda.is_available()返回了false

原因是我cuda是8,pytorch1.0.1匹配的是cuda10,所以只能卸载重新下载低版本。

但我卸载又失败,比较暴躁的直接把环境删了重新建了一个,再重新装pytorch0.4.0就可以啦~

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄