win10+cuda11.1+anaconda+pytorch+pycharm配置环境

CUDA11.0+Anaconda+pytorch+PyCharm

参考链接：
[https://blog.csdn.net/A496608119/article/details/112855153?utm_medium=distribute.pc_relevant.none-task-blog-baidujs_baidulandingword-1&spm=1001.2101.3001.4242]

一、查看CUDA版本

二、安装CUDA

CUDA下载地址

验证是否安装成功

输入nvcc -V，能看到版本信息即安装成功

三、安装cuDNN

cuDNN下载地址
我下载的是8.0.4

完成以上操作注意环境变量的配置

四、安装Anaconda

官网下载地址
傻瓜式一键安装，注意选择***添加环境变量***

五、安装pytorch CUDA

教程
验证是否安装成功：
在创建的虚拟环境下，运行以下命令：

python
import torch
torch.cuda.is_available()

结果返回True，安装成功。

六、在jupyter添加安装的虚拟环境

参考地址
第一次，首先在终端安装ipykernel
再执行

conda install -n 环境名称 ipykernel 

激活虚拟环境

conda activate 环境名称

将环境写入notebook的kernel中

python -m ipykernel install --user --name 环境名称 --display-name "在jupyter中显示的环境名称"

这样打开jupyter就能看到自己安装的环境了

测试，成功！

七、配置PyCharm

在项目的解释器选对应虚拟环境的python即可

八、问题

pycharm运行程序也没有问题
但在jupyter notebook导入torch出错


python版本：3.6
numpy版本：1.19.5
setuptools版本：52.0.0

解决办法：

pip uninstall -y numpy  

pip uninstall -y setuptools

pip install setuptools

pip install numpy

setuptools版本：57.4.0
numpy:1.19.5
(我再也不想装环境了–)