Ubuntu的python3.8虚拟环境安装tensorflow==1.15.0

python3.8下直接pip install tensorflow-gpu==1.15.0安装的话会报错:

 可以使用以下命令来安装:

pip install --upgrade pip
pip install nvidia-pyindex
pip install nvidia-tensorflow[horovod]
pip install nvidia-tensorboard==1.15

测试使用:

python
import tensorflow as tf
tf.__version__

解释一下: 

  1. pip install --upgrade pip: 这个命令用于升级pip工具本身。pip是Python的软件包管理器,用于安装和管理Python软件包。通过运行该命令,可以确保使用最新版本的pip。

  2. pip install nvidia-pyindex: 这个命令用于安装nvidia-pyindex软件包。nvidia-pyindex是一个用于加速pip在NVIDIA GPU上安装Python软件包的索引服务器。它提供了更快的下载速度和更可靠的连接,以提高在NVIDIA GPU环境中构建机器学习模型所需的依赖项的安装效率。

  3. pip install nvidia-tensorflow[horovod]: 这个命令用于安装nvidia-tensorflow软件包和horovod插件。nvidia-tensorflow是NVIDIA专为GPU优化的TensorFlow版本,提供了性能优化和硬件加速功能。horovod是一个用于并行化和分布式训练的框架,可以与nvidia-tensorflow一起使用,以加快训练速度并提高扩展性。

  4. pip install nvidia-tensorboard==1.15: 这个命令用于安装特定版本的nvidia-tensorboard软件包。nvidia-tensorboard是NVIDIA优化的TensorBoard版本,用于可视化和监控深度学习模型的训练过程和性能。通过指定版本号为1.15,安装TensorFlow 1.15兼容的nvidia-tensorboard。

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值