windows10环境下安装深度学习环境anaconda+pytorch+CUDA+cuDDN

 

步骤零：安装anaconda、opencv、pytorch（这些不详细说明）。复制运行代码，若是没有报错，说明已经能够了。不过大几率不行，个人会报错提示AssertionError: Torch not compiled with CUDA enabled。说明须要安装CUDA，或者安装的pytorch版本是不带CUDA的版本，须要按照如下步骤操做。

步骤一：安装CUDA

步骤二：安装cuDDN

步骤三：测试运行代码

附：电脑不支持CUDA或者不想用gpu加速深度学习的

 

 

安装CUDA

这就是用来调用gpu的工具，进行高效并行计算

打开控制面板->NVIDIA控制面板->帮助->系统信息->组件，查看NVCUDA.DLL版本（有些电脑可能不支持CDUA，解决方法在文章最后）

 

进入这个网站下载对应版本的CUDA

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

 

 

下载之后选择自定义安装，选择所有组件安装。注意记下这里的安装目录。

 

安装cuDDN

能够理解为cuda的一个补丁，用来加速深度学习的一些运算的，特意针对深度学习进行优化了（我作的时候没有安装，应该也能够）

进入这个网站下载对应版本的CUDA（须要注册）

https://developer.nvidia.com/rdp/cudnn-download

下载以后，

(1)解压：会生成cuda/include、cuda/lib、cuda/bin三个目录；

(2)分别将cuda/include、cuda/lib、cuda/bin三个目录中的内容拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5（CUDA安装目录）对应的include、lib、bin目录下便可。

(3)将bin所在的目录添加到环境变量 PATH 中，“此电脑”→“高级系统设置”→“环境变量”→“系统变量”→“path”→“编辑”→“新建”加入该路径便可。

 

测试torch可否用gpu

import torch

print(torch.cuda.is_available())

输出True便可。

或者

import torch

print(torch.version.cuda)

输出CUDA的版本号便可。

此时运行你的深度学习程序应该就能够成功啦。

 

若是以前都作好了，却输出False或None则说明你安装的pytorch版本不对，多是不带CUDA的版本。

进入https://pytorch.org/ 选择正确的CUDA版本进行下载。

 

这里我是以前下载了一个None版本的pytorch，卸载之后再用conda命令下载CUDA10.0版本的，结果发现仍是不行。因而我改在pycharm中用pip安装命令安装pytorch，最后解决问题。

 

电脑不支持CUDA或者不想用gpu加速深度学习的小伙伴

改这三个地方就好啦，即把用到CUDA的地方都注释掉

红字是改以前的，绿字是改以后的

 

 

原文出处：https://www.cnblogs.com/lxy764139720/p/11413112.html