win10+pytorch 安装踩坑记录(anaconda+pycharm gpu版)

因为之前电脑安装过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则安装成功在这里插入图片描述

发布了125 篇原创文章 · 获赞 56 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/Nicht_sehen/article/details/102687794