Install failed: Command '['which', 'g++ -std=gnu++0x']' returned non-zero exit status 1

[pekinghk]

pytorch: 1.7.0
cuda:10.1
g++:5.4

install failed:
Traceback (most recent call last):
File "setup.py", line 29, in
cmdclass={'build_ext': BuildExtension})
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/site-packages/setuptools/init.py", line 153, in setup
return distutils.core.setup(**attrs)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/core.py", line 148, in setup
dist.run_commands()
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/dist.py", line 955, in run_commands
self.run_command(cmd)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/dist.py", line 974, in run_command
cmd_obj.run()
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/command/build.py", line 135, in run
self.run_command(cmd_name)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/cmd.py", line 313, in run_command
self.distribution.run_command(command)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/dist.py", line 974, in run_command
cmd_obj.run()
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/site-packages/setuptools/command/build_ext.py", line 79, in run
_build_ext.run(self)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/command/build_ext.py", line 339, in run
self.build_extensions()
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 653, in build_extensions
build_ext.build_extensions(self)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/command/build_ext.py", line 448, in build_extensions
self._build_extensions_serial()
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/command/build_ext.py", line 473, in _build_extensions_serial
self.build_extension(ext)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/site-packages/setuptools/command/build_ext.py", line 202, in build_extension
_build_ext.build_extension(self, ext)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/distutils/command/build_ext.py", line 533, in build_extension
depends=ext.depends)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 482, in unix_wrap_ninja_compile
with_cuda=with_cuda)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 1214, in _write_ninja_file_and_compile_objects
check_compiler_abi_compatibility(compiler)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 252, in check_compiler_abi_compatibility
if not check_compiler_ok_for_platform(compiler):
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/site-packages/torch/utils/cpp_extension.py", line 212, in check_compiler_ok_for_platform
which = subprocess.check_output(['which', compiler], stderr=subprocess.STDOUT)
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/subprocess.py", line 356, in check_output
**kwargs).stdout
File "/search/odin/huangkai/anaconda3/envs/yolo/lib/python3.6/subprocess.py", line 438, in run
output=stdout, stderr=stderr)
subprocess.CalledProcessError: Command '['which', 'g++ -std=gnu++0x']' returned non-zero exit status 1.