Apex安装

最近学习Swintransformer的时候,需要安装apex,官方安装指南。
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
可是按照如上的安装,出现如下错误:
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 10.1.

上述的意思是:cuda与Pytorch的版本不匹配,于是将上述的安装命令的后面部分去掉了,即
pip install -v

Apex安装最先出现在Python成神之路

版权声明:
作者:Mr李
链接:https://www.techfm.club/p/25636.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>