前提：已经安装好相关显卡驱动
Pytorch安装
参考网址：pytorch.org/get-started…
1.创建环境指定python版本
2.安装pytorch以及cuda
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
tf1.15环境安装
1. 安装cuda和cudnn
tf1对应的cuda和cudnn版本
www.tensorflow.org/install/sou…
具体cuda版本 ≤ nvidia-smi命令右上角提示的最高兼容cuda版本
conda install cudatoolkit=10.0 cudnn=7.6.5
2. 安装tensorflow以及keras
pip install tensorflow-gpu==1.15 keras==2.3.1 -i pypi.douban.com/simple/
3.验证安装正确
import tensorflow as tf
print(tf.version)
print(tf.test.is_gpu_available())
