〇、PyCharm和Anaconda安装

PyCharm下载地址

Anacondax下载地址

略略略(一直下一步即可)

如果要同时使用TensorFlow 12.x,请使用CUDA11.x;或者根据实际情况选择

一、查看显卡驱动版本

在终端内执行下面的命令:

nvidia-smi

e4755b4e-bd01-a656-2d21-8e53ffd38ba1.png

可以看到当前安装的Driver Version: 546.17CUDA Version: 12.3,后续的步骤均需要根据此版本进行操作。

在执行后续步骤前也可以去英伟达官网,或者使用软件更新一下显卡驱动(精简安装即可)。

二、安装CUDA Toolkit

CUDA Toolkit和相应的驱动程序版本表

34080260-a22f-0325-ed6c-3e70f0ca2fe2.png

先前查询到Driver Version为546.17,满足最后一列的所有CUDA最低版本要求。此时可以选择CUDA 12.3 Update 1版本的CUDA Toolkit。

CUDA Toolkitx下载地址

选择对应的版本下载即可,我这里点击CUDA Toolkit 12.3.1进行下载。

5ebf91a7-37be-50d3-655d-32935a037e91.png

安装时一直下一步即可。

bcf548b8-ff47-dbcc-8495-38432855ad31.png

三、安装cuDNN

cuDNN下载地址

23263a07-86c9-5213-0952-34c645bccbd9.png

进入cuDNN下载站点,选择最新版本的cuDNN,然后找到和自己的CUDA对应的cuDNN下载即可(需要进行注册登录)。

选择最新的cuDNN v8.9.5,因为先前查到的CUDA Version是12.3,所以可以下载for CUDA 12.x版本。

下载完成后将解压后的文件目录(bin、include、lib)移动至NVIDIA GPU Computing Toolkit目录下(例如我本地是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.3):

251b166f-fdb9-9ebf-acec-4b02d91beb55.png

四、查询对应的Pytorch-CUDA版本

Pytorch版本选择网站

bebcb068-0db1-5be1-b9e7-d253d0591582.png根据之前查询到的CUDA Version: 12.3进行选择,并复制相应的命令:

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

五、创建Conda环境并安装PyTorch-CUDA

在Anaconda Navigator界面下创建一个环境,例如pytorch-cuda

5e0c5f21-a418-0369-f5e2-86bf1aced692.png86afc0c0-66a0-eac4-f7b7-529410c607a4.png

环境创建完成后,点击Open Terminal,

Snipaste_2023-11-24_11-55-31.png

然后在Terminal界面输入上一步复制的命令,然后等待完成(包较大,需要较长时间)。

f9812d9c-806d-bcc1-d886-eea02df50e3f.png

六、验证CUDA是否可用

点击Open with Python

Snipaste_2023-11-24_11-55-31.png

然后执行以下代码:

import torch
torch.__version__
torch.cuda.is_available()
torch.zeros(1).cuda()

显示True表示有可用的CUDA,即安装完成。

77c8026d-7a46-c8e7-f427-3835e9925451.png