强制重启指令
reboot -nf
查看CUDA版本1
nvidia-smi
查看CUDA版本2
nvcc -V
nvcc --version
版本一和版本2查看到的版本可能不同,原因参见nvcc与nvidia-smi CUDA版本不同
那么问题来了:我该如何选择与CUDA版本匹配的Pytorch呢?
结论是,我们应该选择与nvcc --version对应的CUDA版本匹配的Pytorch.
原因参见这个博文
强制重启指令
reboot -nf
查看CUDA版本1
nvidia-smi
查看CUDA版本2
nvcc -V
nvcc --version
版本一和版本2查看到的版本可能不同,原因参见nvcc与nvidia-smi CUDA版本不同
那么问题来了:我该如何选择与CUDA版本匹配的Pytorch呢?
结论是,我们应该选择与nvcc --version对应的CUDA版本匹配的Pytorch.
原因参见这个博文