Pytorch cuda版本选择(高效简洁版)
简而言之
Pytorch cuda版本选择 只需要低于cuda驱动版本即可，cuda驱动版本查看命令是nvidia-smi, nvcc -V 是runtimeapi版本可以不用管
1.只要看cuda驱动版本
安装pytorch 选择cuda版本，只要看你电脑cuda驱动版本即可。
2.选择依据
pytorch中cuda版本只要不高于你电脑cuda驱动版本即可
3.cuda驱动版本方法
nvidia-smi
简单解释
1.原因是cuda 两个版本
一个是驱动版本(nvidia-smi可查)，另一个是runtime api版本（nvcc -v 查的），
2.pytorch的cuda版本只需要低于你的cuda驱动版本即可
原文地址：https://blog.csdn.net/binbinczsohu/article/details/143505636
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！