【发布时间】:2019-09-13 00:08:51
【问题描述】:
我使用虚拟环境在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:
【问题讨论】:
-
TensorFlow 1.12 需要 CUDA 9.0,TensorFlow 2.0 需要 CUDA 10.0。您将需要升级到 CUDA 10。(对于支持 CUDA 10 的 TensorFlow 1.x 版本,可能使用 TensorFlow 1.13.1)
标签: python tensorflow pycharm tensorflow2.0