1查看python对应的pytorch版本
pytorch的github官方文档:
https://github.com/pytorch/vision#installation
只要根据自己的python版本,选择torch就行,torchvision在之后用pytorch的官网的命令可以直接确定(第二步会讲)
近期pytorch版本:
往期pytorch版本:
2查看CUDA版本
输入命令nvcc -V
nvcc -V
如图,我的CUDA版本为11.1
pytorch的github官方文档:
https://github.com/pytorch/vision#installation
只要根据自己的python版本,选择torch就行,torchvision在之后用pytorch的官网的命令可以直接确定(第二步会讲)
近期pytorch版本:
往期pytorch版本:
输入命令nvcc -V
nvcc -V
如图,我的CUDA版本为11.1