3090 运行 tensorflow 1.X,不用docker
最近需要跑个旧代码,不知道为啥,3090 采用docker运行tensorflow1.x 版本,巨慢无比发现一个巨好用的方法:不需要重装cuda,或者安装dockerconda create --name TF1.15 python=3.6conda activate TF1.15pip install nvidia-pyindexpip install nvidia-tensorflow[hor
·
最近需要跑个旧代码,不知道为啥,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了,开心~~~~
参考:
更多推荐
已为社区贡献1条内容
所有评论(0)