win11+RTX3060 cuda cudnn pytorch 安装配置


1、cuda

CUDA Toolkit 11.1.1

网址:CUDA Toolkit 11.1 Update 1 Downloads | NVIDIA Developer

 下载安装即可;

2、cudnn

?cuDNN v8.1.1 (Feburary 26th, 2021), for CUDA 11.0,11.1 and 11.2?

网址:cuDNN Archive | NVIDIA Developer

3、pytorch

pytorch1.9.0

网址:Previous PyTorch Versions | PyTorch

 即以下命令:

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

相关