PyTorch简易安装

PyTorch简易安装

查看电脑CUDA版本

安装anacoda环境

下载anacoda，直接安装

配置环境：

conda create -n pytorch python==3.7 

代码意思：安装环境名 pytorch ， pyrhon版本3.7

查看安装的环境

conda info -e

切换环境

conda activate pytorch

安装pytorch

进入pytorch官网（https://pytorch.org/get-started/previous-versions/）***需要科学上网
找到与电脑cuda相近的版本（我的电脑cuda是11.2的 安装11.3的也成功了）

将相应代码直接复制到anaconda中，即可安装成功

import torch
torch.cuda.is_available()

检测是否安装成功