PyTorch2.0 环境搭建详细步骤(Nvidia显卡)

Step 1 、���看显卡驱动版本

Step2、下载CUDA 11.7 或者11.8（我自己用的这个）也行,稍后我会贴出来版本匹配对应表

CUDA Toolkit - Free Tools and Training | NVIDIA Developer

Step3、下载CUDNN

cuDNN 9.0.0 Downloads | NVIDIA Developer

Step4、安装anconda（尽量使用这种方式安装python环境，坑会少一点）

https://www.anaconda.com/download

Step5、配置环境变量

Step6、执行命令安装PyTorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia