安装CUDA以及pytouch

本文介绍了如何在CSDN博客上下载不同CUDA版本(如11.8和12.1),通过NVIDIA控制面板确定需求,安装对应版本的cuDNN,并指导如何为PyTorch(包括gpu版本和Python版本)及xformers库选择合适的pip安装命令。
摘要由CSDN通过智能技术生成

安装CUDA 以及cuDNN

https://blog.csdn.net/qq_35930739/article/details/128167167

cuda 各个版本下载, 怎么确定cuda版本呢,简单说就是 NVIDIA控制面板- 系统信息 查看所需CUDA版本

https://developer.nvidia.com/cuda-toolkit-archive

安装对应的pytouch版本,注意是gpu版本、以及python版本

https://pytorch.org/get-started/locally/

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

可选 xformers

cuda 11.8 version

pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118

cuda 12.1 version

pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121

  • 11
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值