我要在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的安装目录下,直接覆盖安装即可完成

posted on 2023-03-10 00:44  黑逍逍  阅读(78)  评论(0编辑  收藏  举报