Cuda安装
在官网中选择11.7的版本下载CudaToolkit(本地安装2.5G)
双击点开exe后按步骤选择安装
使用自定义安装
若第一次安装就全选,如果之前安装过就只选择CUDA
等待安装结束
在cmd中使用这个代码检查安装是否成功,同时检查安装的版本的信息
nvcc --version;
PyTorch安装
在conda中新建一个python环境,查看PyTorch官网,选择安装的版本
conda中激活对应的环境,输入这个语句
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
等待下载和安装,完成之后pytorch就安装完毕了