D:\software\anaconda3\envs\pytorch_gpu\lib\site-packages\torch\cuda\__init__.py:143: UserWarning:
NVIDIA GeForce RTX 3070 Laptop GPU 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 3070 Laptop GPU 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))
原因:
win10系统 rtx3070,必须安装cuda11以上版本,对应的pytorch是1.11版本
先安装cuda11.3
https://developer.nvidia.com/cuda-toolkit-archive
然后安装对应版本的cudnn8.2
https://developer.nvidia.com/cudnn
安装pytorch
直接运行指令进行安装
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
如果想下载下来本地安装,则下载地址为
https://download.pytorch.org/whl/torch_stable.html
下载版本为gpu版的11.3
卸载之前安装的pytorch
第一步:卸载当前的pytorch库
在命令框中输入 pip list,查看所卸载的pytorch库是否在列表中,
第二步:在cmd中输入 pip uninstall torch