我要在windows 运行pytorch
下载时候,需要cuda的版本
一、查看电脑上有没有cuda
二、下载cuda
https://blog.csdn.net/qq_40968179/article/details/128996692
1. 打开英伟达控制面板
2. 帮助
3. 得到nvidia能驱动的最大版本
4. 安装cuda 【上面文档】
踩坑1:
5. 安装cudnn
这两个是一起的。
深度学习本质上就是训练深度卷积神经网络。尽管我们已经可以用cuda使得显卡来完成并行计算任务,但是所有的操作还是比较底层的、复杂的。是否在cuda之上有一个专门用于深度神经网络的SDK库来加速完成相关特定的深度学习操作,答案就是cudnn
cuDNN是用于深度神经网络的GPU加速库
https://developer.nvidia.com/cudnn
下载之后,解压缩,将CUDNN压缩包里面的bin、include、lib文件直接复制到CUDA的安装目录下,直接覆盖安装即可完成