win10中安装cuda和TensorFlow-gpu的坑

1.首先查看电脑GPU上cuda的版本

nvcc-V

如果此处提示 :nvcc 不是内部或外部命令,也不是可运行的程序或批处理文件。 说明电脑没有装cuda,需要到nvidia官网去下载cuda,下载时需要注意cuda和TensorFlow-gpu的版本一定要匹配具体请参开TensorFlow官网文件https://www.tensorflow.org/install/source_windows
本人用的是下面的配置

win10中安装cuda和TensorFlow-gpu的坑
win10中安装cuda和TensorFlow-gpu的坑
win10中安装cuda和TensorFlow-gpu的坑

; 2. 安装CUDA

去英伟达官网下载对应版本的cuda,下载完成后先查看电脑是否已经安装了部分cuda程序,如果有需要先卸载(某些程序卸载需要重启电脑),卸载完成后即可以进行安装,首先尝试能不能直接安装,操作流程如下,依次按照提示安装即可。

win10中安装cuda和TensorFlow-gpu的坑
安装过程中可能存在的一些问题:
[En]

Some problems that may exist during installation:

  1. 程序安装失败
    失败后查看C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2目录下是否有Visual Studio Intergration 目录,然后再看下安装报错的程序是不是Nsight compute,如果是的话再次重新安安装将这两个程序取消勾选再安装即可, 安装完成后执行nvcc -V查看是否安装成功
  2. 执行Python代码时报没有发现一系列动态库,例如:
Could not load dynamic library 'cublas64_11.dll'; dlerror: cublas64_11.dll not found

这时先查看C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin目录下是否有当前动态库,如果没有需要可以到https://www.dll-files.com/search/网站去找。
如果出现cublas64_11.dll, cudart64_110.dll找不到需要讲C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin对应的dll文件复制一份到c盘system32下。如图

win10中安装cuda和TensorFlow-gpu的坑
最后,再Python编译器中运行测试代码,显示成功!!
import tensorflow as tf

print(tf.config.list_physical_devices('GPU'))

win10中安装cuda和TensorFlow-gpu的坑

Original: https://blog.csdn.net/qq_43614355/article/details/121519965
Author: qq_43614355
Title: win10中安装cuda和TensorFlow-gpu的坑

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/496420/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球