1.首先要安装 CUDA 和 cuDNN
CUDA:
下载地址:
CUDA Toolkit Archive | NVIDIA Developer
cuDNN
下载地址:
cuDNN Archive | NVIDIA Developer
具体安装步骤可参考:
Windows10系统下CUDA和cuDNN安装教程_Angeldream123的博客-CSDN博客_cudnn怎么安装
安装cuda和cuDNN 完成后,安装pytorch.
1. 可以先安装anaconda:
2. conda 安装完成后使用conda安装pytorch
可在pytorch 官网选择相应的版本得到安装命令:
https://pytorch.org/get-started/locally/#windows-pip
eg:
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
本机的cuda 和cuDNN 版本要高于 pytorch 中cudatoolkit的版本