在Colab尝试复现一篇神经网络论文的时候,遇到了报错:
查找资料发现应该是cuda、torch、torchvision版本不对应。 查看torch的cuda版本
查看机器的cuda版本
发现并不对应,去官网找到符合自己要求的版本https://pytorch.org/get-started/previous-versions/ 按照要求重新按照:
|
在Colab尝试复现一篇神经网络论文的时候,遇到了报错:
查找资料发现应该是cuda、torch、torchvision版本不对应。 查看torch的cuda版本
查看机器的cuda版本
发现并不对应,去官网找到符合自己要求的版本https://pytorch.org/get-started/previous-versions/ 按照要求重新按照:
|