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


去英伟达官网下载对应版本的cuda,下载完成后先查看电脑是否已经安装了部分cuda程序,如果有需要先卸载(某些程序卸载需要重启电脑),卸载完成后即可以进行安装,首先尝试能不能直接安装,操作流程如下,依次按照提示安装即可。
安装过程中可能会存在的一些问题:
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下。如图
最后,再Python编译器中运行测试代码,显示成功!!
import tensorflow as tf
print(tf.config.list_physical_devices('GPU'))
