使用本教程前，默认您已经安装并配置好了python3以上版本

1. 去官网下载匹配的Cuda

Cuda下载地址
当前最高版本的Cuda是12.1
我安装的就是这个版本

小提示：自定义安装可以只选择安装Cuda Runtime。Nvidia全家桶不必全部安装。把全家桶全部安装完直接系统盘占了6G，很大的。

2. 安装pytorch

Pytorch下载地址

我之前使用的是pip install torch (2.0.0版本)
这样安装的torch是直接运行在CPU上的
想要使用GPU版本需要使用对应的cuda版本

尽管pytorch官网上目前只给出了11.8的Cuda支持，但是社区明确表明了兼容高版本Cuda。

上图列出了我本地的torch相关库
可以看见这是一个混乱的构成（一团糟），
torch是cpu版本的
torchaudio是gpu版本的
torchvision是cpu版本的

可以用torch.cuda.is_available()的方法检验自己的torch是不是GPU版本

所以我决定直接全部卸载，重新安装GPU版本的torch全家桶

pip uninstall torch torchvision torchaudio

然后执行安装
这是本篇文章编写时最新版本的安装方法

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

3. 检验是否安装成功

使用pip list查看安装的版本是否正确

执行torch.cuda.is_available() 返回True说明已经在使用GPU版本了

4. (扩展)安装transformer

我之前的版本是4.27.2

但是我在启动huggingface的GLM项目时遇到了动态模块加载的报错。
已经有人给官方提了issue

建议使用4.26.1版本，比较稳定。

pytorch安装GPU版本 (Cuda12.1)教程相关推荐

pytorch 安装 GPU版本 1030也能装
已经安装了anaconda 核实是否为英伟达显卡,我电脑是入门级1030,在英伟达算力表上没有,但是也可以装. nvidia-smi 1.安装CUDA CUDA Toolkit Archive | N ...
Pytorch：入门指南和 PyTorch 的 GPU版本安装(非常详细)
Pytorch: 入门指南和 PyTorch 的 GPU版本安装(非常详细) Copyright: Jingmin Wei, Pattern Recognition and Intelligent S ...
AI 图片截取、ffmpeg使用及安装, anaconda环境，图片标注(labelme)，模型训练(yolov5)，CUDA+Pytorch安装及版本相关问题
AI 图片截取(ffmpeg), anaconda环境,图片标注(labelme),模型训练(yolov5),CUDA+Pytorch安装及版本相关问题一.截取有效图片录制RTSP视频脚本 #!/ ...
tensorflow运行环境linux,在ubuntu或者min运行环境下安装gpu版本的tensorflow
转载请注明出处:blog.csdn.net/sproll 本文描述在ubuntu系列操作系统上安装gpu版本的tensorflow的过程. 0,准备工作 BIOS中关闭板载显卡,显示器接在nvidia ...
英伟达3060Ti安装GPU版本TensorFlow2.X
查看Python与TensorFlow对应版本安装GPU版本的TensorFlow的时候,我们需要考虑的一个问题是Python版本与TensorFlow版本的对应关系,可以参考下面这个链接: Pyt ...
windows10 安装gpu版本TensorFlow脑壳疼自我记录
写文章时间为:2018.10.26 显卡为NVIDIA GeForce RTX 2080Ti(刚出没有多久),系统为window10 放在开始想说的是,以下是我安装过程,仅是自我总结,仅供参考. 再之 ...
常识——windows的tensorflow安装gpu版本,cuda算力只有3.0也可以
文章目录问题 windows安装tensorflow_gpu版本检查显卡是否支持cuda 查看自己的显卡型号去官网查看是否支持cuda 检查显卡的cuda算力按照GPU算力3.0的电脑配置进行 ...
安装 GPU 版本的 tensorflow 完整版本
前言: 之前安装的 CPU 版本的 tensorflow 一直出问题,索性就直接安装 GPU 版本的 tensorflow 了(有了GPU 就不能浪费). 安装过程: 1)看自己有无 GPU,找到对应 ...
安装GPU版本的pytorch（解决pytorch安装时默认安装CPU版本的问题）保姆级教程
1.安装anaconda anaconda官网:www.anaconda.com 2.修改下载源为清华源由于pytorch的服务器在国外,直连下载的话很慢,所以选用清华镜像源下载 # 添加清华镜像 ...

pytorch安装GPU版本 (Cuda12.1)教程

1. 去官网下载匹配的Cuda

2. 安装pytorch

3. 检验是否安装成功

4. (扩展)安装transformer

pytorch安装GPU版本 (Cuda12.1)教程相关推荐

最新文章

热门文章