torch和cuda的辛酸史

可以找到不同的对应版本:
https://download.pytorch.org/whl/torch_stable.html

千万注意,同一个版本的torch\torchvision 都会对应多个cuda,所以一定要根据对应的cuda去下载;最好是根据官网提供的链接,公司或者单位提供的内部环境可能不是对应的cuda!!!一定要check好这一点。也可以本地下载后,再上传到服务器上。

https://pytorch.org/get-started/locally/----官网

你可能感兴趣的:(项目学习)