1.创建环境
conda create -n pytorch-gpu python=3.7
2.激活环境
activate pytorch-gpu
3.在激活的环境中使用命令查看cuda的版本
nvidia-smi
结果如下:
4.在pytorch官网(https://pytorch.org/)查找对应版本的pytorch安装命令进行安装
使用该命令进行安装
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
5.测试是否安装成功
True即为成功安装。