python - TF 2.0 GPU cofingure，如何在pycharm中创建另一个虚拟环境后配置tensorflow-gpu 2.0和cuda、cudnn

Question

我已经使用虚拟环境在Pycharm中安装了TensorFlow 1.12，GPU可以正常工作。当我创建另一个虚拟环境并安装 TensorFlow 2.0 Alpha 时，GPU 不再工作。

错误:

Could not dlopen library 'libcudart.so.10.0'; dlerror: libcudart.so.10.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-9.0/lib64:

Answer 1

我也遇到了同样的问题。我通过将以下命令添加到“.bashrc”文件来修复它。

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64/

系统配置:

Ubuntu 16.04 LTS
Tensorflow GPU 2.0beta1
Cuda 10.0
cuDNN 7.6.0 for Cuda 10.0

我使用 conda 来配置我的系统。