win10安装CUDA9.0+CUDANN+Anaconda3+PYTORCH1.x

最新推荐文章于 2024-07-23 14:10:27 发布

qq_37172182

最新推荐文章于 2024-07-23 14:10:27 发布

阅读量1.4k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Pytorch 文章标签： CUDA Pytorch GPU

本文链接：https://blog.youkuaiyun.com/qq_37172182/article/details/99537757

Pytorch 专栏收录该内容

21 篇文章

订阅专栏

本文详细介绍在Windows 10环境下，使用NVIDIA GeForce 940M显卡配置CUDA 9.0和Pytorch的过程，包括软件下载、环境变量设置、Anaconda安装、Pytorch安装及jupyter notebook设置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

当前神经网络方法的训练和预测越来越依赖于GPU并行运算，因此在win10上配置CUDA+Pytorch，记录配置过程。

硬件平台：

win10(64位), NVIDIA Geforce 940M

软件下载：

1：CUDA9.0，安装时注意加载路径（环境变量）

官网地址：CUDA9.0下载

2：CUDANN，对应CUDA9.0版本的CUDANN

官网地址：CUDANN7.6下载

3:解压CUDANN

将对应bin/cudann64_7.dll, include/cudnn.h, lib/x64/cudnn.lib分别加载到CUDA包对应的bin, include, lib/x64文件中；

打开控制台，输入：

nvcc -V

显示如下：

4：安装Anaconda3.5.2.0(对应Python3.6)，清华镜像下载

官网地址：Anaconda3.5.2.0下载

安装环境时添加python路径，正确安装后，打开控制台显示结果（注意有anaconda后缀）：

5：安装pytorch

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

6：设置jupyter notebook默认地址

找到jupyter notebook属性页：

将目标值设为：

C:\ProgramData\Anaconda3\python.exe C:\ProgramData\Anaconda3\cwp.py C:\ProgramData\Anaconda3 C:\ProgramData\Anaconda3\python.exe C:\ProgramData\Anaconda3\Scripts\jupyter-notebook-script.py "设置的文件路径"

起始位置与“设置的文件路径”相同

代码补全:

pip install jupyter_contrib_nbextensions
jupyter contrib nbextension install --user
pip install--user jupyter_nbextensions_configurator
jupyter nbextensions_configuratorenable--user

出现结果（勾选Hinterland）：

7：测试