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

Question

I have installed TensorFlow 1.12 in Pycharm using a virtual environment, the GPU can work well. When I create another virtual environment and install TensorFlow 2.0 Alpha , the GPU doesn't work anymore.

The error:

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

I had the same issue. I fixed it by adding the below command to the ' .bashrc ' file.

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

System configuration:

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

I used conda to configure my system.