网站首页 全球最实用的IT互联网站!

人工智能P2P分享Wind搜索发布信息网站地图标签大全

当前位置:诺佳网 > 人工智能 > 自动驾驶 >

win10中安装cuda和TensorFlow-gpu的坑

时间:2021-11-24 17:33

人气:

作者:admin

标签:

导读:1.首先查看电脑GPU上cuda的版本nvcc-V如果此处提示 :nvcc 不是内部或外部命令,也不是可运行的程序或批处理文件。说明电脑没有装cuda,需要到nvidia官网去下载cuda,下载时需要注意cuda和...

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

nvcc-V

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

2. 安装CUDA

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

  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下。如图
在这里插入图片描述
最后,再Python编译器中运行测试代码,显示成功!!

import tensorflow as tf

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

在这里插入图片描述

温馨提示:以上内容整理于网络,仅供参考,如果对您有帮助,留下您的阅读感言吧!
相关阅读
本类排行
相关标签
本类推荐

CPU | 内存 | 硬盘 | 显卡 | 显示器 | 主板 | 电源 | 键鼠 | 网站地图

Copyright © 2025-2035 诺佳网 版权所有 备案号:赣ICP备2025066733号
本站资料均来源互联网收集整理,作品版权归作者所有,如果侵犯了您的版权,请跟我们联系。

关注微信