炼丹环境torch

查看主机支持的cuda版本

控制面板

查看支持的最大版本

下载支持的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()