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 查的),