ubuntu 20.04 安装 huggingface transformers 环境

1. 安装 cuda

大多数新发布的大语言模型使用了较新的 PyTorch v2.0+ 版本,Pytorch 官方认为 CUDA 最低版本是 11.8 以及匹配的 GPU 驱动版本。详情见Pytorch官方 如下图:
在这里插入图片描述

1.1 下载 cuda

cuda 12.1 官方网站
在这里插入图片描述
下载:

$wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run

1.2 安装 cuda

$sudo sh cuda_12.1.1_530.30.02_linux.run

然后,接受协议
在这里插入图片描述
由于已经安装显卡驱动,选择不安装,然后光标到 install 处,点击回车
在这里插入图片描述
同意覆盖原有的 /usr/local/cuda
查看cuda:

$ ll /usr/local/cuda
lrwxrwxrwx 1 root 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值