安装nvidia-pyindex 和nvidia-tensorflow[horovod] 需要的nvidia包的whl 已经传到百度网盘
nvidia_tensorflow-1.15.5+nv22.04
CUDA12
在A100的显卡上运行TensorFlow1.15.5成功。
conda create --name python=3.8
在新环境创建后,激活,然后安装这些nvidia-tensorflow依赖的whl,metadata的要求在nvidia_tensorflow-1.15.5+nv22.04-4387458-cp38-cp38-linux_x86_64.whl\nvidia_tensorflow-1.15.5+nv22.04.dist-info\METADATA里,先按照这里面的要求pip安装whl,没有whl的就直接conda install(tensorflow-estimitor可以先跳过)然后再
pip install nivida-pyindex
pip install nvidia-tensorflow[horovod]
安完之后没有keras,则pip install keras -i https://pypi.tuna.tsinghua.edu.cn/simple
安装不易,一起努力。
链接:https://pan.baidu.com/s/1-rrAnKjlEnYLYlvthFjC4Q
提取码:fig5