时间:2023-05-06 13:13:04 阅读:9726 作者:2617
安装pytorch的时候,需要知道电脑的cuda版本。 很多博客中介绍的华丽复杂的命令很难使用,甚至会进入小偷的洞穴。
这里可以用一个命令解决。 JJDD。
nvidia-smi
说明我的cuda版本是11.4,然后参考pytorch公式匹配就可以了。
版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。