python-pytorch-cuda下载参考(自用,推荐有用链接)

版本考虑顺序(仅供参考,考虑配置难度,个人倾向):cuda+pytorch-->python,记得配置在虚拟环境,可以配多个。

  • 前提准备:anaconda,cuda(可根据需求自行下载新版本)

Linux安装Anaconda_怎么看anconda有没有安装成功 linux-优快云博客

  • 查看CUDA版本:

win+R;cmd;

nvidia-smi
  • pytorch和cuda对应版本官方参考和下载链接:

Previous PyTorch Versions | PyTorch

  • pytorch和python对应版本官方github参考:

pytorch/vision: Datasets, Transforms and Models specific to Computer Vision

  • conda在虚拟环境下载对应cuda版本的pytorch等镜像源问题可参考:

anaconda最新国内镜像源更换解决清华源失效问题_conda清华镜像不可用-优快云博客

  • 详细介绍相关指令含义可参考:

基于conda安装pytorch---一看就会超稳定安装步骤详解【多次踩坑实践最终成功版!!!】【适用于WIN系统】-优快云博客

其他官网

pytorch_lighting对应pytorch:

Versioning Policy — PyTorch Lightning 2.5.0.dev documentation

更新:

使用离线安装方法:如果服务器网络真的很慢,可以考虑在本地下载 wheel 文件,然后上传到服务器进行安装。

访问PyTorch官方wheel文件下载页面:

https://download.pytorch.org/whl/torch_stable.html

选择正确的版本:(下图示例,适用于服务器安装)

上传到服务器(可考虑xftp上传,需要更新文件夹)

cd ~/pytorch_wheels #这里放文件夹名字(‘pytorch_wheels’)

!进入虚拟环境!(代码省略)

pip install torch-1.13.1+cu116-*.whl
pip install torchvision-0.14.1+cu116-*.whl
pip install torchaudio-0.13.1+cu116-*.whl

验证

# 在Python中验证GPU是否可用
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name(0))

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值