最近需要跑个旧代码,不知道为啥,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了,开心~~~~

参考:

https://www.pugetsystems.com/labs/hpc/How-To-Install-TensorFlow-1-15-for-NVIDIA-RTX30-GPUs-without-docker-or-CUDA-install-2005/

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐