下载安装CUDA11.3
CUDA是Nvidia推出的只能用于自家GPU的并行计算框架。只有安装这个框架才能够进行复杂的并行计算
cudnn是一个GPU 加速的深度神经网络基元库**,能够以高度优化的方式实现标准例程(如前向和反向卷积、池化层、归一化和激活层)
详细了解可参考大神的文章
查看对应版本的CUDA,这部非常关键!!!请一定要重视,避免之后多次重装。
(1)首先查看自己电脑GPU版本
win+R打开cmd,输入nvidia-smi
(2)查看自己对应的版本
CUDA下载链接:developer.nvidia.com/cuda-toolki…
接下来,我们来下载这个CUDA,这里选择的版本不能高于你的显卡驱动里面那个版本号,由于我的是12.1,且系统是win10,所以我这里选择的是12.1
安装
这里与其他安装方法不同的是,我没有安装 cudnn,也没有配置 path ,但是同样也安装成功。
打开 Anaconda prompt 命令,输入命令:
nvcc -V
下载GPU版本下的pytorch和pytorchvision
下载 pytorch 的下载链接为:
里面的文件是cpu开头的是 CPU 版本,cu开头的才是我们要下载的GPU版本。
这里可参考 torch 和 torchvision 的对应图,以免下错对应版本,这里我选择的是红色圈内的,根据自己需求下载
下载后放在同一目录下,我在d盘新建文件夹 CUDA-python,放在了E:\CUDA-pytorch下:
下载完成后切换到下载目录,使用pip命令安装。
检验,输入命令:
pip list
验证
- import torch
- torch.cuda.is_available()
- True True代表安装成功