因为之前电脑安装过gpu版本的tf,所以我还是使用的之前安装的cuda和cudnn,我的cuda是9.1版本
安装教程参考
https://blog.csdn.net/kan2281123066/article/details/93789303
这个安装教程写的很好,不过我还是不可避免的踩到了一些莫名奇妙的坑,在这里记录一下
1. pytorch 官网无cuda9.1版本
我直接下载的9.0版本,亲测可用
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
如果换源了需要去掉后面的 -c pytorch
conda install pytorch torchvision cudatoolkit=9.0
2.配置pycharm时找不到路径
在anaconda prompt中输入
conda env list
选择路径时点击下面图标会出现隐藏文件夹
3.配置到pycharm环境时报错:Please specify a different SDK name
原因: 虚拟环境名字重复,删掉重新配置就好
4. 测试时运行报错:ImportError: numpy.core.multiarray failed to import
卸载掉之前下过的numpy,重新安装。这一步我卸载重装了很多次=_=,最后终于可以运行成功了
卸载:
pip uninstall numpy
重装:
pip install numpy
测试代码:
import torch
print(torch.cuda.is_available())
输出为True则安装成功