CYRZFX的博客

原创 解决5070显卡出现pytorch不支持sm_120问题

本文介绍了为NVIDIA 50系显卡安装PyTorch的方法。首先需从PyTorch官网获取Nightly预览版的CUDA 12.8版本安装链接，选择与系统匹配的最新torch和torchvision版本（后者日期需晚一天）。然后创建对应的Anaconda Python环境，通过pip安装下载的whl文件。最后通过代码验证安装是否成功，若输出包含"sm_120"和True则表示环境配置正确。该方法解决了50系显卡不兼容当前PyTorch正式版的问题。