1.下载Anaconda
2.打开Anaconda Prompt,显示(base)表明安装成功.
3.管理环境,Prompt中输入conda create -n pytorch python=3.8.8,接着输入y安装必要的包
4.激活pytorch环境,输入 conda activate pytorch
5.显示已安装的包,输入 pip list
6.下载pytorch, Start Locally | PyTorch 选择稳定版,
安装命令 conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
7.查看CUDA版本,cmd中输入 nvidia-smi
8.查看pytorch是否安装成功,以及查看是否支持cuda
pytorch中输入 python, 回车
输入 import torch(没有报错证明安装成功)
输入 torch.conda.is_available(), 返回True证明可使用CUDA