1、英伟达驱动安装或更新:https://www.nvidia.cn/geforce/drivers/
2、进入torch官网,选择对应驱动版本,使用conda安装方法,会自动安装cuda和cudnn
3、检查pytorch 和tensorflow安装gpu是否可用
import torch
print(torch.cuda.is_available())
import tensorflow as tf
print(tf.test.is_gpu_available())
返回true代表可用gpu