【问题标题】:TF 2.0 GPU cofingure, how to configure the tensorflow-gpu 2.0 and cuda, cudnn after create another virtual environment in pycharmTF 2.0 GPU coingure,如何配置tensorflow-gpu 2.0和cuda,cudnn在pycharm中创建另一个虚拟环境后
【发布时间】:2019-09-13 00:08:51
【问题描述】:

我使用虚拟环境在Pycharm中安装了TensorFlow 1.12GPU可以正常工作。 当我创建另一个虚拟环境并安装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


【解决方案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 来配置我的系统。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2020-02-03
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2016-10-22
    • 2017-10-25
    • 2022-10-15
    相关资源
    最近更新 更多