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成神之路。
共有 0 条评论