Anaconda配置pytorch
- CUDA11.0+Anaconda+pytorch+PyCharm
-
- 一、查看CUDA版本
- 二、安装CUDA
-
- 验证是否安装成功
- 三、安装cuDNN
- 四、安装Anaconda
- 五、安装pytorch CUDA
- 六、在jupyter添加安装的虚拟环境
- 七、配置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 numpypip uninstall -y setuptoolspip install setuptoolspip install numpy
【安装问题|win10+cuda11.1+anaconda+pytorch+pycharm配置环境】setuptools版本：57.4.0
numpy:1.19.5
(我再也不想装环境了–)
推荐阅读
- pytorch|使用pytorch从头实现多层LSTM
- SG平滑轨迹算法的原理和实现
- 推荐系统论文进阶|CTR预估 论文精读(十一)--Deep Interest Evolution Network(DIEN)
- Python专栏|数据分析的常规流程
- pytorch|YOLOX 阅读笔记
- Python|Win10下 Python开发环境搭建(PyCharm + Anaconda) && 环境变量配置 && 常用工具安装配置
- Python绘制小红花
- Pytorch学习|sklearn-SVM 模型保存、交叉验证与网格搜索
- OpenCV|OpenCV-Python实战（18）——深度学习简介与入门示例
- python|8. 文件系统——文件的删除、移动、复制过程以及链接文件