Pytorch 安装 apex 疑难杂症解决方案

最新推荐文章于 2024-03-06 17:17:02 发布

原创

最新推荐文章于 2024-03-06 17:17:02 发布 · 1w 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

在尝试安装apex时遇到cudatoolkit与pytorch.cuda版本不匹配的问题。原有cudatoolkit为10.0，而pytorch.cuda为9.0.176。通过重新安装pytorch未改变cuda版本。搜索解决方案，尝试从官网寻找10.0版本的pytorch.cuda未果，后发现conda安装命令，但实际使用的是pip。最终在CUDA10的相关网址找到适用于pip安装的torch1.1-cuda100-python3.5版本，安装后成功安装apex。

RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries.  Pytorch binaries were compiled with Cuda 9.0.176.

从描述可以看出是cudatoolkit与pytorch.cuda的版本不匹配问题，看了一下自己的cudatoolkit==10.0，pytorch.cuda却是9.0.176

所以重新安装了pytorch，发现pytorch.cuda版本没变。

去网上搜方案，一堆都是去官网找10.0的pytorch.cuda版本，但是官网上并没有找到