文章目录[隐藏]
处理CUDA安装过程中的问题汇总(tensorflow.pytorch)
1.首先找到自己的控制面板看一下自己的配置
2.确认自身系统支持最大的CUDA
3.前往 https://developer.nvidia.com/cuda-toolkit 去下载。在安装的时候一定要自定义安装,否则将会安装很多无用的东西。安装的选项,可以选择不更新驱动程序
4.选择自己下载的CUDA执行安装
执行以下操作:
5.下载与CUDA版本匹配的CUDNN
官网下载:https://developer.nvidia.com/rdp/cudnn-archive
解压文件夹,将解压后的文件夹下的文件拷贝到cuda安装目录下,与之相对应的文件夹下。
6.配置环境变量
将自己的CUDA位置添加到里面并保存退出
7.至此我们的CUDA就安装完成了!
tensorflow对应CUDA网址:https://tensorflow.google.cn/install/source_windows
pytorch对应CUDA网址: https://pytorch.org/get-started/previous-versions/
版权声明:本文为CSDN博主「XiaoChao_AI」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_38008528/article/details/118033428
暂无评论