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的坑_第1张图片
win10中安装cuda和TensorFlow-gpu的坑_第2张图片
win10中安装cuda和TensorFlow-gpu的坑_第3张图片

2. 安装CUDA

去英伟达官网下载对应版本的cuda,下载完成后先查看电脑是否已经安装了部分cuda程序,如果有需要先卸载(某些程序卸载需要重启电脑),卸载完成后即可以进行安装,首先尝试能不能直接安装,操作流程如下,依次按照提示安装即可。
win10中安装cuda和TensorFlow-gpu的坑_第4张图片
安装过程中可能会存在的一些问题:

  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的坑_第5张图片
最后,再Python编译器中运行测试代码,显示成功!!

import tensorflow as tf

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

win10中安装cuda和TensorFlow-gpu的坑_第6张图片

你可能感兴趣的:(python,自动驾驶,机器学习)