|
UserWarning:
NVIDIA GeForce RTX 3060 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.
If you want to use the NVIDIA GeForce RTX 3060 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
训练遇到上面的错误，原因是30系显卡目前只支持cuda11.0版本系列，需要根据显卡算力重新配置cuda环境
若想知道自己的pytorch版本是否匹配
在命令行输入python
impot torch
torch.zeros(1).cuda()
|