Pytorch安装使用国内镜像方式

操作方法参考网址: https://blog.csdn.net/gg864461719/article/details/112259806

pytorch官方网址: https://pytorch.org/

pytorch不是最新的版本官方网址: https://pytorch.org/get-started/previous-versions/

pytorch国内镜像网址: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

测试安装的cuda版本torch

测试安装的cuda版本torch是否可以使用,使用以下命令测试可能会更好:

import torch
torch.zeros(1).cuda()

如果输出如下则说明可以使用cuda版本的torch:
在这里插入图片描述
如果报错是GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.那么一定是cuda版本的问题了。安装cuda版本不对。

解决办法:

降低CUDA版本:

1.在运行pytorch程序的时候报这样的错:
GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
具有CUDA功能的GeForce RTX 3080 sm_86与当前Pythorch安装不兼容。当前Pythorch安装支持CUDA功能sm_37sm_50 sm_60 sm_61 sm_70 sm_75 compute_37。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值