1 前期准备
创建conda环境
conda create -n Pytorch python=3.6
进入conda环境
conda activate Pytorch
查看cuda版本
nvcc -V
输出:Cuda compilation tools, release 10.0, V10.0.130
表示CUDA版本是10.0
2 安装步骤
(1)进入官网,查找对应版本命令
(2)执行安装
pip install torch torchvision torchaudio