目录
- 1. 更新最新的显卡驱动
- 2. 安装CUDA
- 3. 安装cuDNN
- 4. 安装pytorch
1. 更新最新的显卡驱动
-
打开NVIDA更新驱动的官网地址
-
根据下图的选择,记得
Windows驱动程序类型
要选标准
,如图
-
点击搜索,下面就会列出一大堆的历史驱动,选择第一个,也就是最新的,当前是472.12,点开,再点下载,再点同意并下载,建议用EagleGet
-
下载完毕后,双击安装,会提示选择一个解压缩的目录,建议选一个空闲空间大一点的盘来存放,这个安装完毕后会自动删除的
-
等解压缩完毕后,会自动检查系统是否可以安装这个驱动,等检查完毕后,选择
NVIDA图形驱动程序
,上面的NVIDIA显卡驱动和GeForce Experience
,会多安装GeForce Experience,给可怜的C盘更多的负担,不建议选择
-
同意并继续后,选择
精简
,下一步
-
等待安装完成,关闭就行了,
建议重启电脑一下
-
右键点击桌面空白处,选择
NVIDIA 控制面板
,打开后,点击左下角的系统信息
,查看驱动程序版本
是否是安装的472.12
2. 安装CUDA
- 打开cmd,输入命令:
nvidia-smi
,会列出当前的驱动版本和支持的CUDA的最高版本,如果提示命令不存在,建议用软件everything
,搜索nvidia-smi.exe
,然后自行把目录添加到环境变量里,或者直接进入nvidia-smi.exe
的目录里再运行命令
- 右上角显示的11.4就是当前电脑能够支持CUDA的最高版本,可以向下兼容,去CUDA的官方地址下载,选择低于可支持版本的,比如我选择了下图这个
- 点进去后,根据电脑情况选择
- 再往下拉一点,使用下载工具下载,建议用EagleGet
- 下载完毕后,双击安装,会提示选择一个解压缩的目录,建议选一个空闲空间大一点的盘来存放,这个安装完毕后会自动删除的
- 等解压缩完毕后,会自动检查系统是否可以安装这个驱动,等检查完毕后,同意并继续
- 选
自定义
,下一步
- 点开CUDA,找到
Visual Studio Integration
,取消勾选,如图所示
- 跟CUDA同级别的也都取消勾选,如图所示,下一步
- 选择安装位置,因为我C盘紧张,就选择安装到其他盘了,下一步,开始安装
- 等待安装完成,下一步,关闭就行了,
建议重启电脑一下
- 打开cmd,输入命令:
nvcc -V
,能正常输出版本信息等,就说明安装成功了
- (可选),做一下GPU带宽测试,用软件
everything
,搜bandwidthTest.exe
,找到文件所在地址,然后打开cmd,输入命令:bandwidthTest.exe
,看到结果有输出pass,说明测试成功
- (可选),获取CUDA可使用的GPU的信息,用软件
everything
,搜deviceQuery.exe
,找到文件所在地址,然后打开cmd,输入命令:deviceQuery.exe
,看到结果有输出pass,说明测试成功,根据结果也可知,集显是不能被CUDA操控的。
3. 安装cuDNN
- 先注册一个NVIDIA的开发者账号,打开地址,根据提示验证并注册
- 打开cuDNN的官方下载地址,需要根据已经安装的CUDA的版本选择
- 点开后,找到Windows的,点开,下载
- cuDNN其实称不上安装,只需要将下载下来的压缩包解压后,将对应文件夹的文件放到CUDA安装路径下的对应文件夹里即可(cuDNN可以认为是CUDA的补充,可以加速深层神经网络的速度)
- 直接把上图的所有都复制,然后打开CUDA的文件夹,直接无脑粘贴就行了,如果找不到,用软件
everything
,搜CUDA_Toolkit_Release_Notes.txt
或者nvcc.exe
,找到CUDA的安装文件夹,也是一样有bin、include、lib
的才是
4. 安装pytorch
-
打开cmd,输入命令:
pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117
,torch有2g多,如果速度够快的话,就不用管,等它自动下完(不建议用国内源加速,据说有坑,会在没有gpu版本的时候直接安装cpu版本)
-
如果速度实在太慢的话,把下面这3个链接用下载工具先下载到本地,然后用python的whl安装包方法进行一一安装
-
https://download.pytorch.org/whl/cu117/torch-1.13.0%2Bcu117-cp310-cp310-win_amd64.whl
-
https://download.pytorch.org/whl/cu117/torchvision-0.14.0%2Bcu117-cp310-cp310-win_amd64.whl
-
https://download.pytorch.org/whl/cu117/torchaudio-0.13.0%2Bcu117-cp310-cp310-win_amd64.whl
-
检查是否安装成功,打开cmd,输入命令:
python
,进入python的运行环境 -
输入
import torch
,稍等一下等它加载 -
再输入
print(torch.cuda.is_available())
,得到结果True
,棒!
-
理论上,上面安装成功的话,下面这些就不用看了~
-
官网地址,可以尝试官网推荐的方法(没试过),
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
-
如果官网首页找不到对应的cuda版本,可以进这个链接,通过ctrl+f搜索,但是神奇的是,cuda 11.4版本没有找到,最后想到直接用whl的方法试试,就前面写的,成功了!!!
enjoy!!!