PyCharm和Tensorflow

gojuced7  于 2023-05-01  发布在  PyCharm
关注(0)|答案(1)|浏览(207)

我有一个带有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.

物理设备列表显示为:

[]

我做错了什么?

mqkwyuun

mqkwyuun1#

如果你有一个nvidia gpu,tensorflow有一个bug,所以这不是你的错。

相关问题