首先 创建conda环境
conda create -n pytorch python=3.6
进入新建环境
conda activate pytorch
安装 无cuda加速 的版本 pytorch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
安装完成后,检测是否安装成功
import torch
安装 cuda 10.1 的版本 pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
安装完成后,检测是否安装成功
import torch
torch.cuda.is_available()
返回结果是True表示pytorch已经安装完成。
也可以pytorch官网查找特定的安装指令