如何解决关于 TensorFlow 安装要求的困惑
我现在正在维护一个使用 tensorflow-gpu-1.11
的遗留项目。根据此link,此包与某些特定版本的 gcc
、CUDA 和 cuDNN
Version Python version Compiler Build tools cuDNN CUDA
...
tensorflow_gpu-1.11.0 2.7,3.3-3.6 GCC 4.8 Bazel 0.15.0 7 9
...
因此,我认为我需要
- 第 1 步:将系统编译器设置为
gcc-4.8
,可能使用sudo update-alternatives --config ...
- 第 2 步:安装 CUDA 9.0 和关联的 cuDNN 7
- 第 3 步:创建一个虚拟环境并
pip install tensorflow_gpu==1.11.0
但是,安装此软件包的另一种方法是在 conda
环境中调用以下命令
conda install -c anaconda tensorflow-gpu=1.11.0
其中与官方要求相反,尝试安装更高版本的cudatoolkit
和cudnn
,如下图
The following NEW packages will be INSTALLED:
...
cudatoolkit anaconda/linux-64::cudatoolkit-9.2-0
cudnn anaconda/linux-64::cudnn-7.2.1-cuda9.2_0
...
tensorflow anaconda/linux-64::tensorflow-1.11.0-gpu_py36h9c9050a_0
tensorflow-base anaconda/linux-64::tensorflow-base-1.11.0-gpu_py36had579c0_0
tensorflow-gpu anaconda/linux-64::tensorflow-gpu-1.11.0-h0d30ee6_0
...
我不确定我是否正确理解了安装要求。如果没有,希望有人能帮助走上正轨。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。