炼丹环境torch
本帖最后由 lideshan 于 2024-8-6 13:49 编辑#
## 查看主机支持的cuda版本
###控制面板
查看支持的最大版本
## 下载支持的cuda版本
### 11.8这个版本比较稳定
(https://developer.nvidia.com/cuda-toolkit-archive)
一路默认安装即可
!(https://img.lideshan.top/i/2024/07/30/66a884b1b92fa.png)
## 下载对应cuda版本的cudnn版本
[各个cudnn版本](https://developer.nvidia.com/cuda-toolkit-archive)
一路默认安装即可
!(https://img.lideshan.top/i/2024/07/30/66a884b07d849.png)
## 下载对应cuda的torch版本
(https://pytorch.org/get-started/locally/)
```bash
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
```
!(https://img.lideshan.top/i/2024/07/30/66a884b2d0060.png)
## 测试炼丹的环境安装是否安装成功
```python
import torch
torch.__version__
torch.cuda.is_available()
```
!(https://img.lideshan.top/i/2024/07/30/66a884b163dca.png) 感谢分享 用pytorch如果不是自己编译，不需要手动下载cuda还有cudnn那些东西
pip装了就能用了，不信你试试 楼上是对的，使用pytorch提供的命令安装torch的时候会自动安装对应版本的cuda，不需要自己再去安装一次cuda
我也纳闷为啥要装cuda 根据3楼的方法可以优化安装步骤，谢谢{:301_993:} 本帖最后由 ABuSiDeLuoYin 于 2024-8-6 17:48 编辑
我也不明白为什么要装CUDA,我在其他地方看的CosyVoice安装教程也让安装CUDA和CUDNN,但是我去github上看官方教程并没有这一步(项目也使用了torch),然后我就没装CUDA也能跑CosyVoice 试试看,感谢楼主 艾莉希雅 发表于 2024-8-6 14:24
用pytorch如果不是自己编译，不需要手动下载cuda还有cudnn那些东西
pip装了就能用了，不信你试试
默认是跑cpu的，gpu版本都是不同cu版本对应包很鸡肋 ase1993 发表于 2024-8-10 13:37
默认是跑cpu的，gpu版本都是不同cu版本对应包很鸡肋
不啊，这个主要是看你装了啥版本的pytorch
一楼的命令行加了这个
--index-url https://download.pytorch.org/whl/cu118
装的就是cuda11.8兼容版本
页:
[1]
2