pytorch学习笔记001-安装

安装gpu版本

• 查看英伟达显卡的驱动型号
  

• 安装对应版本的cuda toolkit（cuda不能超驱动型号） 前往官网

• 安装对应版本的cuDNN加速模块 前往官网

• 安装pytorch，安装时选择对应的cuda版本 前往官网
  1.使用conda建立python3.8的虚拟环境
  2.激活虚拟环境输入官网获取的含有版本信息的安装代码（使用清华镜像源安装时删除 -c pytorch，否则默认从pytorch官网下载）

  

• 测试gpu版本pytorch是否安装成功

import torch
torch.cuda.is_available()