较新版本的python3.8/cuda 11如果想要使用tf1.15,一种方法是重新编译,或者去下载其他人编译好的二进制whl。但是重新编译和找whl都比较麻烦,老黄为了让新显卡用户用上tf1.15,和谷歌合作维护了一个tf1.15的库。另一种方法是使用docker,但是docker用起来稍麻烦。

库的地址为:

https://github.com/NVIDIA/tensorflow

安装命令:

# 推荐在虚拟环境下操作(非必要)
# python3 -m virtualenv venv
# source venv/bin/activate
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__

参考:
https://segmentfault.com/a/1190000040354322?sort=votes

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐