栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

win10中安装cuda和TensorFlow-gpu的坑

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

win10中安装cuda和TensorFlow-gpu的坑

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 FilesNVIDIA GPU Computing ToolkitCUDAv11.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 FilesNVIDIA GPU Computing ToolkitCUDAv11.2bin目录下是否有当前动态库,如果没有需要可以到https://www.dll-files.com/search/网站去找。
如果出现cublas64_11.dll, cudart64_110.dll找不到需要讲C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.2bin对应的dll文件复制一份到c盘system32下。如图

最后,再Python编译器中运行测试代码,显示成功!!

import tensorflow as tf

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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/589983.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号