安装pytorch成功但cuda不可用

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

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

但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就可以啦~

猜你喜欢

转载自www.cnblogs.com/woshizhizhang/p/10701867.html