我有一个带有tensorflow的conda环境(2.12)和cudnn包安装。我按照tensorflow网站上的安装说明进行了操作。当我从命令行运行python时,我可以准确地看到我的GPU:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
在PyCharm-Professional中运行相同的测试时,我遇到了困难。我有以下几点。bashrc文件:
export CUDNN_PATH=/home/<userid>/miniconda3/envs/AlphaChess_Lambda/lib/python3.10/site-packages/nvidia/cudnn
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib
export TF_CPP_MIN_LOG_LEVEL=2
我已经在PyCharm中使用这些环境变量编辑了我的运行配置。然而,当我的代码启动时,tensorflow报告说没有GPU:
2023-04-30 10:46:33.887206: I tensorflow/tsl/cuda/cudart_stub.cc:28] Could not find cuda drivers on your machine, GPU will not be used.
物理设备列表显示为:
[]
我做错了什么?
1条答案
按热度按时间mqkwyuun1#
如果你有一个nvidia gpu,tensorflow有一个bug,所以这不是你的错。