Ubuntu 安装 tensorflow-gpu 1.15.0

cond create -n tf python=3.6

pip install tensorflow-gpu==1.15.0

conda install cudatoolkit=10.0

conda install cudnn=7.6.5

#检查gpu是否有用

import tensorflow as tf 

tf.test.is_gpu_available() #输出为True则正常

环境打包

conda env export > test.yaml

conda env create -f test.yaml

pip list --format=freeze > requirements.txt

pip install -r requirements.txt