ubuntu18.04 cuda11.0 虚拟环境cuda10.0,pytorch1.2,tensorflow1.15 keras2.3.1

各个版本:

Python 3.7

conda create -n dl_env python=3.7

pytorch 1.2.0 + torchvision 0.4.0 + cuda10.0.130

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch -i https://pypi.tuna.tsinghua.edu.cn/simple

cudnn 7.6.5

conda install -c anaconda cudnn

protobuf 3.20.0

pip uninstall protobuf

pip install protobuf==3.20.0

tensorflow-gpu 1.15.0

conda install tensorflow-gpu=1.15.0

也可以下载了在安装,地址:

https://files.pythonhosted.org/packages/bc/72/d06017379ad4760dc58781c765376ce4ba5dcf3c08d37032eeefbccf1c51/tensorflow_gpu-1.15.0-cp37-cp37m-manylinux2010_x86_64.whl

keras-2.3.1

 pip install keras==2.3.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

其他的安装过程中缺啥补啥,如果有失败的看详细信息进行安装或者更新版本补充

更多refer:

ubuntu16.04配置CUDA10.0.130+cuDNN7.4.1+Anaconda5.2.0+Python3.6+Tensorflow2.0.0+Keras2.3.0+Pytorch1.2.0_今年不吃饭...的博客-CSDN博客

你可能感兴趣的:(tensorflow,pytorch,深度学习)