论坛
潜水/灌水快乐,沉淀知识,认识更多同行。
ToB圈子
加入IT圈,遇到更多同好之人。
朋友圈
看朋友圈动态,了解ToB世界。
ToB门户
了解全球最新的ToB事件
博客
Blog
排行榜
Ranklist
文库
业界最专业的IT文库,上传资料也可以赚钱
下载
分享
Share
导读
Guide
相册
Album
记录
Doing
搜索
本版
文章
帖子
ToB圈子
用户
免费入驻
产品入驻
解决方案入驻
公司入驻
案例入驻
登录
·
注册
只需一步,快速开始
账号登录
立即注册
找回密码
用户名
Email
自动登录
找回密码
密码
登录
立即注册
首页
找靠谱产品
找解决方案
找靠谱公司
找案例
找对的人
专家智库
悬赏任务
圈子
SAAS
IT评测·应用市场-qidao123.com
»
论坛
›
人工智能
›
人工智能
›
4060显卡+LLaMA-Factory微调LLM情况准备——(一)安装c ...
4060显卡+LLaMA-Factory微调LLM情况准备——(一)安装cuda
悠扬随风
金牌会员
|
2024-8-16 16:11:48
|
显示全部楼层
|
阅读模式
楼主
主题
984
|
帖子
984
|
积分
2952
本地设置
系统
:win10
**硬件:**i5-12490f+RTX4060
安装原因:
最近实验在本地对开源LLM举行Fine-tune的时候,用到了LLaMA-Factory,在运行的时候,弹出未检测到GPU,后来才发现,是忘记安装cuda等情况导致的
LLaMA-Factory: https://github.com/hiyouga/LLaMA-Factory
本地测试方法:
GPU:
import torch
print(torch.cuda.is_available()) # 是否可以用gpu False不能,True可以
print(torch.cuda.device_count()) # gpu数量, 0就是没有,1就是检测到了
复制代码
Cuda:
nvcc -V
复制代码
出现以下内容,阐明你本机安装过cuda了
安装过程:
安装前先简单相识下什么是cuda
CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够办理复杂的计算问题。
安装前先安装VS Community,由于不安装的话,直接安装Cuda会报错
下载链接: https://my.visualstudio.com/Downloads?PId=8228
下载前需要注册登岸账号,目前可以下载的版本可能只有2022版的,不过问题不大,也可以用
安装过程是傻瓜式操作,在选择工作负荷的时候,选C++桌面端开发
安装完成后,登录自己的账号(下载前注册的账号),然后关闭即可
去Nvidia 官网,下载CUDA:
https://developer.nvidia.com/cuda-12-2-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local
由于llama-factory要求的版本是11.8-12.2的,索性就下载安装一个12.2的版本
下载的文件大约3G左右
整个安装过程,也是傻瓜式操作,自定义安装中,可以看到安装CUDA的选项
背面一只下一步就好,安装完成后需要设置情况变量,一般默认安装路径都是一样的
设置好后,测试安装是否成功,假如出现下图,看到Build cuda_12.2表现安装成功了
这里大家可能有疑问,nvidia-smi实行后,右上角也会出现一个CUDA Version: 12.5, 这两个cuda版本有什么区别,其实CUDA有runtime api 和 driver api, nvcc -V
表现的是runtime api, nvidia-smi表现的是 driver api, runtime api 是CUDA Toolkit install 安装的, driver api是GPU driver installer安装的, 以是这两个东西安装上并没有什么太大的关系
唯一需要注意的是,一般nvidia-smi的版本需要大于 nvcc -V
表现的版本,像我安装的12.5>12.2,使用起来没有什么大问题
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
本帖子中包含更多资源
您需要
登录
才可以下载或查看,没有账号?
立即注册
x
回复
使用道具
举报
0 个回复
倒序浏览
返回列表
快速回复
高级模式
B
Color
Image
Link
Quote
Code
Smilies
您需要登录后才可以回帖
登录
or
立即注册
本版积分规则
发表回复
回帖并转播
回帖后跳转到最后一页
发新帖
回复
悠扬随风
金牌会员
这个人很懒什么都没写!
楼主热帖
(8) PyQt 设计并实现【工厂扫码装箱系 ...
Java Long类parseLong()方法具有什么功 ...
王心凌再次爆火,为了防止收费,我连夜 ...
【Web前端】HTML详解(上篇)
IDEA: 如何导入项目模块 以及 将 Java ...
详解kubernetes五种暴露服务的方式 ...
命题逻辑等值演算
Oracle ORA-10917: TABLESPACE GROUP c ...
Uniapp进行APP打包——iOS 系统 ...
互联网官方协议标准(rfc5000) ...
标签云
AI
运维
CIO
存储
服务器
浏览过的版块
网络安全
物联网
IOS
快速回复
返回顶部
返回列表