Scalsol / mega.pytorch

Memory Enhanced Global-Local Aggregation for Video Object Detection, CVPR2020
Other
566 stars 115 forks source link

INSTALL.md Problem #120

Open ComeOnComeOnTurnYourRadioOn opened 1 year ago

ComeOnComeOnTurnYourRadioOn commented 1 year ago

Hello, I failed to follow the mega.pytorch.INSTALL.md due to the CUDA version being incompatible with the Pytorch version. I exactly followed the INSTALL.md step by step, and I did install pytorch=1.3.0 and cudatoolkit=10.0 successfully. But, when I tried to install apex, some errors just popped up. Below is the error message.

` torch.version = 1.3.0

Compiling cuda extensions with nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2020 NVIDIA Corporation Built on Mon_Nov_30_19:08:53_PST_2020 Cuda compilation tools, release 11.2, V11.2.67 Build cuda_11.2.r11.2/compiler.29373293_0 from /usr/local/cuda/bin

Traceback (most recent call last): File "setup.py", line 178, in check_cuda_torch_binary_vs_bare_metal(CUDA_HOME) File "setup.py", line 37, in check_cuda_torch_binary_vs_bare_metal

I followed this link and the answer is to comment this check. I also followed this answer, but another even weird error message was poped up.

cc1plus: warning: command line option ‘-Wstrict-prototypes’ is valid for C/ObjC but not for C++ Traceback (most recent call last): File "setup.py", line 788, in <module> extras_require=extras, File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/__init__.py", line 87, in setup return distutils.core.setup(**attrs) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/core.py", line 185, in setup return run_commands(dist) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/core.py", line 201, in run_commands dist.run_commands() File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/dist.py", line 968, in run_commands self.run_command(cmd) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/dist.py", line 1217, in run_command super().run_command(command) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/dist.py", line 987, in run_command cmd_obj.run() File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/command/install.py", line 74, in run self.do_egg_install() File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/command/install.py", line 123, in do_egg_install self.run_command('bdist_egg') File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/cmd.py", line 319, in run_command self.distribution.run_command(command) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/dist.py", line 1217, in run_command super().run_command(command) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/dist.py", line 987, in run_command cmd_obj.run() File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/command/bdist_egg.py", line 165, in run cmd = self.call_command('install_lib', warn_dir=0) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/command/bdist_egg.py", line 151, in call_command self.run_command(cmdname) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/cmd.py", line 319, in run_command self.distribution.run_command(command) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/dist.py", line 1217, in run_command super().run_command(command) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/dist.py", line 987, in run_command cmd_obj.run() File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/command/install_lib.py", line 11, in run self.build() File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/command/install_lib.py", line 112, in build self.run_command('build_ext') File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/cmd.py", line 319, in run_command self.distribution.run_command(command) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/dist.py", line 1217, in run_command super().run_command(command) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/dist.py", line 987, in run_command cmd_obj.run() File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/command/build_ext.py", line 84, in run _build_ext.run(self) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/Cython/Distutils/old_build_ext.py", line 186, in run _build_ext.build_ext.run(self) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/command/build_ext.py", line 346, in run self.build_extensions() File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line 353, in build_extensions build_ext.build_extensions(self) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/Cython/Distutils/old_build_ext.py", line 195, in build_extensions _build_ext.build_ext.build_extensions(self) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/command/build_ext.py", line 466, in build_extensions self._build_extensions_serial() File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/command/build_ext.py", line 492, in _build_extensions_serial self.build_extension(ext) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/command/build_ext.py", line 246, in build_extension _build_ext.build_extension(self, ext) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/command/build_ext.py", line 554, in build_extension depends=ext.depends, File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/setuptools/_distutils/ccompiler.py", line 599, in compile self._compile(obj, src, ext, cc_args, extra_postargs, pp_opts) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line 269, in unix_wrap_compile "'-fPIC'"] + cflags + _get_cuda_arch_flags(cflags) File "/home/wangpengfei/anaconda3/envs/MEGA/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line 991, in _get_cuda_arch_flags raise ValueError("Unknown CUDA arch ({}) or GPU not supported".format(arch)) ValueError: Unknown CUDA arch (8.6) or GPU not supported

BTW, the GPU I am currently using is 3090. I am not too sure if 3090 can support the CUDA version which is less than 11. The cuda drive I am installing is 455.45.01.

May I please ask you if there are any other ways to sort out my problems??

Thank you so much for your help. Looking forward to your response soon! Cheers! :-)

lemyanh2612 commented 1 year ago

I can install this work on GTX1070 and GTX2080, but can not install them on GTX3090.

For apex, You can follow:

  1. git clone https://github.com/NVIDIA/apex.git
  2. cd apex
  3. git checkout e3794f422628d453b036f69de476bf16a0a838ac
  4. python setup.py install