配置CUDA-GPU(CUDA11.6+Torch1.12.1)

本文档详细介绍了CUDA的安装步骤,包括下载、安装路径选择、自定义安装选项以及取消VisualStudioIntegration的勾选。同时,指导如何将CUDA相关文件夹复制到指定路径,并添加环境变量。最后,通过Python的torch库验证CUDA安装是否成功,如果`torch.cuda.is_available()`返回`True`,则表明CUDA安装成功。

1、点击以下链接

百度网盘 请输入提取码

提取码:wx4l --来自百度网盘超级会员V4的分享

2、双击上图中的.exe文件出现以下界面

👉安装路径不要改,点击OK

 2.1 点同意并继续

2.2 这里选自定义安装,并且下一步

2.3 取消勾选CUDA下面的Development下的Visual Studio Intergration选项,然后再下一步

2.4 记住红色框框中的路径,不要去改动,然后下一步,等待安装完成

3、打补丁(打开这个文件夹)

3.1 复制下面三个文件夹并且粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6路径下(前提是你没有改安装路径)

4 添加环境变量

4.1 打开查看高级系统设置-->环境变量

4.2 点新建 然后添加以下四个环境变量

5 验证是否成功

5.1 首先装torch 复制安装命令

5.2 进入cmd然后 粘贴进行安装

5.3 安装成功 就是验证是否成功与cuda一起安装成功

输入 python

import torch

torch.cuda.is_available()

如果输出为True就已经成功,如果不成功就会输出为Flase

test@ubuntugpu:~$ nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2021 NVIDIA Corporation Built on Wed_Jun__2_19:15:15_PDT_2021 Cuda compilation tools, release 11.4, V11.4.48 Build cuda_11.4.r11.4/compiler.30033411_0 test@ubuntugpu:~$ nvidia-smi Fri Aug 15 08:46:29 2025 +-----------------------------------------------------------------------------+ | NVIDIA-SMI 470.256.02 Driver Version: 470.256.02 CUDA Version: 11.4 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |===============================+======================+======================| | 0 Tesla K20m Off | 00000000:00:06.0 Off | Off | | N/A 37C P0 47W / 225W | 0MiB / 5062MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ +-----------------------------------------------------------------------------+ | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | |=============================================================================| | No running processes found | +-----------------------------------------------------------------------------+怎么跑1g内的中国ai模型,我需要详细的全流程部署步骤,包括安装依赖前提条件以及安装模型后怎么测试,核对匹配所有版本以及兼容性关系确保可以适配
08-16
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值