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

win10+pytorch 安装踩坑记录(anaconda+pycharm gpu版)_第1张图片
win10+pytorch 安装踩坑记录(anaconda+pycharm gpu版)_第2张图片
选择路径时点击下面图标会出现隐藏文件夹
win10+pytorch 安装踩坑记录(anaconda+pycharm gpu版)_第3张图片
3.配置到pycharm环境时报错:Please specify a different SDK name
原因: 虚拟环境名字重复,删掉重新配置就好
win10+pytorch 安装踩坑记录(anaconda+pycharm gpu版)_第4张图片
4. 测试时运行报错:ImportError: numpy.core.multiarray failed to import
卸载掉之前下过的numpy,重新安装。这一步我卸载重装了很多次=_=,最后终于可以运行成功了
卸载:

pip uninstall numpy

重装:

pip install numpy

测试代码:

import torch
print(torch.cuda.is_available())

输出为True则安装成功win10+pytorch 安装踩坑记录(anaconda+pycharm gpu版)_第5张图片

你可能感兴趣的:(机器学习,计算机视觉)