最近需要跑个旧代码,不知道为啥,3090 采用docker运行tensorflow1.x 版本,巨慢无比
发现一个巨好用的方法:不需要重装cuda,或者安装docker
conda create --name TF1.15 python=3.6
conda activate TF1.15
pip install nvidia-pyindex
pip install nvidia-tensorflow[horovod]
conda install -c conda-forge openmpi
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/anaconda3/envs/TF1.15/lib/
就可以正常跑 tensorflow1.X了,开心~~~~
参考: