栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

win10系统+RTX3070+cuda11.3+cudnn8.2

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

win10系统+RTX3070+cuda11.3+cudnn8.2

D:softwareanaconda3envspytorch_gpulibsite-packagestorchcuda__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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/929190.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号