[已解决]显卡驱动、cuda、cudnn安装都没问题,却无法启用GPU
Tofloor
poster avatar
joah1
deepin
2020-06-26 01:58
Author
本帖最后由 joah1 于 2020-8-10 19:35 编辑

跑了一个GPU,CPU加速对比程序,明显没有启用GPU计算,状态栏也显示CPU在计算(这点可以确认)

nvidia-smi 显示驱动也没问题,实在是没辙了,求助论坛大佬




用显卡驱动管理器切换开源驱动也失败(可以切换,重启之后失败)

Reply Favorite View the author
All Replies
avatar
joah1
deepin
2020-06-26 02:05
#1
NVIDIA-Linux-x86_64-440.82.run  ---->NVIDIA-Linux-x86_64-450.51.run 重装了一下驱动还是不行
Reply View the author
avatar
muniesa
deepin
2020-06-26 02:42
#2
本帖最后由 muniesa 于 2020-6-25 18:44 编辑

试试源里面的驱动呢 sudo apt install nvidia-driver,另外cuda和cudnn也可以在源里面装,名字叫nvidia-cuda-toolkit和libcudnn7
Reply View the author
avatar
joah1
deepin
2020-08-11 03:28
#3
今天终于解决了困扰我许久的问题,现在把方法分享一下,也感谢大家的帮助第一是cuda版本与tensorflow版本不匹配,tf2.2—>2.0,Python3.8—>3.7,  cuda10.0—cudnn10.0—tensorflow2.0.0—python3.7
第二个问题就是在pycharm执行代码不能启动GPU, 解决方法:
  1. cat ~/.bashrc #查看环境变量
Copy the Code

打开pycharm的setting->python console,添加CUDA_HOME和LD_LIBRARY_PATH环境变量,其中CUDA_HOME 的值就是PATH的值。

再按照同样的方式设置一下 Configuration:

至此,已经添加了环境变量,问题应该可以解决。

Reply View the author