title: 炼丹环境torch
date: 2024-07-30
tags:

• torch
• cuda

• cudnn

  查看主机支持的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()