0
点赞
收藏
分享

微信扫一扫

显卡驱动版本、cuda版本、pytorch版本的匹配查找网站

首先是安装自己的显卡驱动,根据显卡的型号进行选择

https://www.nvidia.cn/Download/index.aspx

查看显卡驱动和cuda的对应版本

Release Notes :: CUDA Toolkit Documentation

查看显卡的计算能力,该计算能力直接与pytorch的版本有关

CUDA GPUs | NVIDIA Developer

关于CUDAGPU算力匹配问题

https://www.csdn.net/tags/MtjakgwsNjE4NjAtYmxvZwO0O0OO0O0O.html

已知cuda版本查看和安装所需要的pytorch版本

Start Locally | PyTorch

举报

相关推荐

0 条评论