炼丹环境torch

lideshan · 发表于 2024-8-6 13:41

本帖最后由 lideshan 于 2024-8-6 13:49 编辑

#

查看主机支持的cuda版本

控制面板

查看支持的最大版本
控制面板.png

下载支持的cuda版本

11.8这个版本比较稳定

cuda各个版本下载
一路默认安装即可

下载对应cuda版本的cudnn版本

各个cudnn版本
一路默认安装即可

下载对应cuda的torch版本

torch版本

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

测试炼丹的环境安装是否安装成功

import torch
torch.__version__
torch.cuda.is_available()

qq66 · 发表于 2024-8-6 13:47

感谢分享

艾莉希雅 · 发表于 2024-8-6 14:24

用pytorch如果不是自己编译，不需要手动下载cuda还有cudnn那些东西
pip装了就能用了，不信你试试

keweiye · 发表于 2024-8-6 14:47

楼上是对的，使用pytorch提供的命令安装torch的时候会自动安装对应版本的cuda，不需要自己再去安装一次cuda

beichen1031 · 发表于 2024-8-6 15:32

我也纳闷为啥要装cuda

Liebesfreud · 发表于 2024-8-6 15:34

根据3楼的方法可以优化安装步骤，谢谢

Zeroive · 发表于 2024-8-6 16:46

希望下一个可以整一个docker镜像

ABuSiDeLuoYin · 发表于 2024-8-6 17:47

本帖最后由 ABuSiDeLuoYin 于 2024-8-6 17:48 编辑

我也不明白为什么要装CUDA,我在其他地方看的CosyVoice安装教程也让安装CUDA和CUDNN,但是我去github上看官方教程并没有这一步(项目也使用了torch),然后我就没装CUDA也能跑CosyVoice

candor · 发表于 2024-8-7 20:27

试试看,感谢楼主

ase1993 · 发表于 2024-8-10 12:37

艾莉希雅发表于 2024-8-6 14:24
用pytorch如果不是自己编译，不需要手动下载cuda还有cudnn那些东西
pip装了就能用了，不信你试试

默认是跑cpu的，gpu版本都是不同cu版本对应包很鸡肋

帐号		自动登录	找回密码
密码			注册[Register]

[学习记录] 炼丹环境torch