GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.

报错如下:
GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
RuntimeError: CUDA error: no kernel image is available for execution on the device

很奇怪的问题,单独安装torch和torchvision会出问题,必须使用pip install torch==1.7.0+cu110 torchvision==0.8.0+cu110 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
这种命令安装。

安装完后在虚拟环境下使用pip list命令查看安装的torch版本是这样的:
请添加图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值