
报错原因:服务器CUDA版本和自己装pytorch 的cuda版本不一致,自己装的是cuda10.0,而服务器上的是cuda9.0。所以修改自己pytorch的cuda版本到9.0。
多一句嘴:自己装cuda9.0的pytorch可以在cuda9.0和cuda10.0的机器上跑。
查看服务器的cuda版本:
nvcc -V

查看自己装pytorch的cuda版本:
python
import torch
torch.version.cuda

本文介绍了如何解决因服务器CUDA版本与本地PyTorch版本不匹配导致的问题,步骤包括检查服务器CUDA版本并调整本地PyTorch版本,确保兼容性。作者提到自行安装cuda9.0的PyTorch可以在两种CUDA环境下运行。

报错原因:服务器CUDA版本和自己装pytorch 的cuda版本不一致,自己装的是cuda10.0,而服务器上的是cuda9.0。所以修改自己pytorch的cuda版本到9.0。
多一句嘴:自己装cuda9.0的pytorch可以在cuda9.0和cuda10.0的机器上跑。
查看服务器的cuda版本:
nvcc -V

查看自己装pytorch的cuda版本:
python
import torch
torch.version.cuda

您可能感兴趣的与本文相关的镜像
PyTorch 2.8
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
1万+
7061
3265
1775
2125
1807

被折叠的 条评论
为什么被折叠?