安装教程 https://blog.csdn.net/RunAtWorld/article/details/124282176 官方英文教程 https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-windows

pytorch需要安装 cuda及cuDNN,这是两个nvidia的包 cuda是开发包 cudnn是

• CUDA toolkit（toolkit就是指工具包）
• cuDNN(cuDNN 是用于配置深度学习使用)

cuda下载地址 https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local cuDNN 下载地址 https://developer.nvidia.com/rdp/cudnn-archive pytorch 官网 https://pytorch.org/

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

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600 wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda-repo-ubuntu2204-12-1-local_12.1.0-530.30.02-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu2204-12-1-local_12.1.0-530.30.02-1_amd64.deb sudo cp /var/cuda-repo-ubuntu2204-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/ sudo apt-get update sudo apt-get -y install cuda