docker支持GPU镜像之tensorflow 1.14.0

该博客详细介绍了如何在服务器上搭建基于CUDA 10.0的GPU环境,包括安装英伟达驱动、Miniconda以及TensorFlow 1.14.0的GPU版本。通过清华大学和北京外国语大学的镜像源加速下载,并提供了测试代码以验证环境是否配置成功。要求读者已经具备NVIDIA驱动和Docker的基础。
部署运行你感兴趣的模型镜像

查找合适的GPU基础镜像

安装支持python运行环境

安装tensorflow

# 腾讯云安装
pip install  tensorflow-gpu==1.14.0 -i https://mirrors.cloud.tencent.com/pypi/simple/
# 删除安装  tensorflow==1.14.0

安装驱动

 conda install cudatoolkit-10.1.243-h6bb024c_0.tar.bz2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
 
 conda install cudnn-7.6.5-cuda10.1_0.tar.bz2  -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
上面安装过程较慢,可以下载下来直接安装

清华下载地址
北外下载地址

# 安装
conda install --use-local cudatoolkit-10.1.243-h6bb024c_0.tar.bz2
conda install --use-local  cudnn-7.6.5-cuda10.1_0.tar.bz2

测试能否使用

测试代码链接

服务器前提要求:
  1. 已安装nvidia
  2. 已安装docker
    如有帮助,请您三连,多谢

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

你遇到的错误: ``` ERROR: tensorflow_gpu-1.14.0-cp36-cp36m-manylinux1_x86_64.whl is not a supported wheel on this platform. ``` 表示你正在尝试安装的 `.whl` 文件 **与当前平台(Python 环境或操作系统)不兼容**。 --- ### 🔍 常见原因分析 #### ✅ 1. **Python 版本不匹配** - 该 `.whl` 文件是为 **Python 3.6** 编译的(`cp36` 表示 CPython 3.6)。 - 如果你的 Python 版本不是 3.6(比如是 3.7、3.8、3.9 等),就会报错。 #### ✅ 2. **操作系统或架构不匹配** - 该 `.whl` 文件是 `manylinux1_x86_64`,表示适用于 Linux 的 64 位系统。 - 如果你使用的是 Windows 或 macOS,或者使用了 32 位 Linux 系统,就无法安装。 #### ✅ 3. **虚拟环境与全局环境混淆** - 可能你在使用 `conda` 环境,但使用的是全局 `pip`。 - 或者你激活了某个虚拟环境,但该环境的 Python 版本与 `.whl` 不匹配。 --- ### ✅ 解决方法 #### ✅ 1. 检查当前 Python 版本 ```bash python --version ``` 确保输出是: ``` Python 3.6.x ``` 如果不是,请使用 `conda` 创建一个 Python 3.6 的虚拟环境: ```bash conda create -n tf114 python=3.6 conda activate tf114 ``` 然后再尝试安装: ```bash pip install tensorflow_gpu-1.14.0-cp36-cp36m-manylinux1_x86_64.whl ``` --- #### ✅ 2. 使用 `pip debug --verbose` 检查当前平台支持的 wheel 格式 ```bash pip debug --verbose ``` 查看输出中的 `Compatible tags`,例如: ``` cp36-cp36m-manylinux_2_17_x86_64 cp36-cp36m-linux_x86_64 ``` 确认你的环境是否支持 `cp36` 和 `manylinux1` 标签。 --- #### ✅ 3. 替代方案:从源码安装 TensorFlow 1.14(适用于 flightmare) 如果你只是为了运行 [flightmare](https://github.com/uzh-rpg/flightmare),可以尝试使用源码安装 TensorFlow 1.14: ```bash pip install tensorflow-gpu==1.14.0 ``` > 注意:这需要你的系统有足够内存(至少 4GB RAM)。 --- #### ✅ 4. 使用 Docker(推荐) 如果你不想折腾本地环境,可以直接使用 TensorFlow 1.14 的官方 Docker 镜像: ```bash docker run -it tensorflow/tensorflow:1.14.0-gpu-py3 ``` > 需要先安装 [Docker](https://docs.docker.com/engine/install/) 和 [NVIDIA Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide/) --- ### ✅ 验证是否安装成功 ```bash python -c "import tensorflow as tf; print(tf.__version__); print(tf.test.is_gpu_available())" ``` 预期输出: ``` 1.14.0 True ``` --- ### ❓
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值