TF 2.0 GPU配置，如何在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 ”文件中來修復該問題。

導出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來配置我的系統。