不能与Pytorch一起使用GPU

时间:2019-12-27 23:24:16

标签: tensorflow machine-learning anaconda artificial-intelligence pytorch

尝试使用Pytorch时,我总是收到此错误。

  

RuntimeError:尝试在CUDA设备上反序列化对象,但是torch.cuda.is_available()为False。如果您在仅CPU的计算机上运行,​​请使用带有map_location = torch.device('cpu')的torch.load将存储映射到CPU。

我使用conda install pytorch torchvision cudatoolkit=10.1 -c pytorch安装了Pytorch。

使用tensorflow,我的GPU运行得很好。

2 个答案:

答案 0 :(得分:1)

您可以通过安装CUDA 10.2(最新版本)来解决此错误,并另外使用以下命令重新安装Pytorch:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

答案 1 :(得分:0)

使用#title-wrapper { display: flex; justify-content: center; align-items: center; font-family: "Aclonica", serif; color: #281c1c; font-size: 6rem; text-transform: uppercase; margin: 3rem 1rem; } li { list-style: none; } #header-img { width: 6rem; padding-right: 1.5rem; } #nav-bar ul { display: flex; flex-direction: column; align-items: center; font-family: "Libre Baskerville", serif; background-color: #990000; color: #e0e0e0; border-top: 1px solid black; border-bottom: 1px solid black; text-transform: uppercase; font-size: 4rem; } #nav-bar ul li { border: 2px solid black; width: 100%; text-align: center; } #nav-bar ul li a { text-decoration: none; color: inherit; padding: 1.4rem; display: block; } #nav-bar ul li:hover { background: #cc3300; } 上的官方安装命令

对于Windows + GPU(对于其他,请检查pytorch.org

pytorch.org