Relation and Difference between CUDA和cuDNN

安装顺序并非唯一, 只要你把正确的东西塞到正确的位置就ok了

0. Install Nvidia Graphics Driver

macOS用户 | 核显用户请跳过此步骤.

Windows出厂自带, 可以使用Geforce Experience更新或者切换.

Linux需要安装闭源 (non-free) 驱动

1. Install Anaconda (miniconda is also available)

1.1 前往https://www.anaconda.com下载并安装anaconda.

1.2 创建指定python版本的环境, 并安装numpy, pandas等packages

2. Install Pytorch

前往https://pytorch.org安装对应版本的Pytorch, Nvidia独显用户请选择cuda版本, amd独显用户请选择ROCm版本, 核显用户请选择cpu版本.

eg. conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

使用conda命令安装cuda版本Pytorch时，会同时安装cudatoolkit和cuDNN, pytorch-cuda=11.7 就是安装cuda的部分, 其主要包含应用程序在使用CUDA相关的功能时所依赖的动态链接库。