炼丹环境torch
...小于 1 分钟
查看主机支持的cuda版本
控制面板
查看支持的最大版本
下载支持的cuda版本
11.8这个版本比较稳定
cuda各个版本下载 一路默认安装即可
下载对应cuda版本的cudnn版本
各个cudnn版本 一路默认安装即可
下载对应cuda的torch版本
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
测试炼丹的环境安装是否安装成功
import torch
torch.__version__
torch.cuda.is_available()
Powered by Waline v3.3.0