My GPU is Ampere A100, the lowest version of cuda it supported is cuda 11.*.  So I cannot directly use pre-built pytorch 1.5.1.
I did the build process as the instruction of officical site. But it output a Make Error.
Building wheel torch-1.5.0a0+3c31d73
-- Building version 1.5.0a0+3c31d73
cmake -GNinja -DBUILD_PYTHON=True -DBUILD_TEST=True -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=/data/xielongfei/pytorch/torch -DCMAKE_PREFIX_PATH=/data/xielongfei/miniconda3/envs/t -DNUMPY_INCLUDE_DIR=/data/xielongfei/miniconda3/envs/t/lib/python3.7/site-packages/numpy/core/include -DPYTHON_EXECUTABLE=/data/xielongfei/miniconda3/envs/t/bin/python -DPYTHON_INCLUDE_DIR=/data/xielongfei/miniconda3/envs/t/include/python3.7m -DPYTHON_LIBRARY=/data/xielongfei/miniconda3/envs/t/lib/libpython3.7m.so.1.0 -DTORCH_BUILD_VERSION=1.5.0a0+3c31d73 -DUSE_NUMPY=True /data/xielongfei/pytorch
-- Performing Test SUPPORT_GLIBCXX_USE_C99
-- Performing Test SUPPORT_GLIBCXX_USE_C99 - Failed
CMake Error at cmake/MiscCheck.cmake:81 (message):
  The C++ compiler does not support required functions.  This is very likely
  due to a known bug in GCC 5 (and maybe other versions) on Ubuntu 17.10 and
  newer.  For more information, see:
  https://github.com/pytorch/pytorch/issues/5229
Call Stack (most recent call first):
  CMakeLists.txt:400 (include)


-- Configuring incomplete, errors occurred!
See also "/data/xielongfei/pytorch/build/CMakeFiles/CMakeOutput.log".
See also "/data/xielongfei/pytorch/build/CMakeFiles/CMakeError.log".
Traceback (most recent call last):
  File "setup.py", line 744, in <module>
    build_deps()
  File "setup.py", line 316, in build_deps
    cmake=cmake)
  File "/data/xielongfei/pytorch/tools/build_pytorch_libs.py", line 59, in build_caffe2
    rerun_cmake)
  File "/data/xielongfei/pytorch/tools/setup_helpers/cmake.py", line 323, in generate
    self.run(args, env=my_env)
  File "/data/xielongfei/pytorch/tools/setup_helpers/cmake.py", line 141, in run
    check_call(command, cwd=self.build_dir, env=env)
  File "/data/xielongfei/miniconda3/envs/t/lib/python3.7/subprocess.py", line 363, in check_call
    raise CalledProcessError(retcode, cmd)
subprocess.CalledProcessError: Command '['cmake', '-GNinja', '-DBUILD_PYTHON=True', '-DBUILD_TEST=True', '-DCMAKE_BUILD_TYPE=Release', '-DCMAKE_INSTALL_PREFIX=/data/xielongfei/pytorch/torch', '-DCMAKE_PREFIX_PATH=/data/xielongfei/miniconda3/envs/t', '-DNUMPY_INCLUDE_DIR=/data/xielongfei/miniconda3/envs/t/lib/python3.7/site-packages/numpy/core/include', '-DPYTHON_EXECUTABLE=/data/xielongfei/miniconda3/envs/t/bin/python', '-DPYTHON_INCLUDE_DIR=/data/xielongfei/miniconda3/envs/t/include/python3.7m', '-DPYTHON_LIBRARY=/data/xielongfei/miniconda3/envs/t/lib/libpython3.7m.so.1.0', '-DTORCH_BUILD_VERSION=1.5.0a0+3c31d73', '-DUSE_NUMPY=True', '/data/xielongfei/pytorch']' returned non-zero exit status 1.

This is /CMakeError.log
Performing C++ SOURCE FILE Test SUPPORT_GLIBCXX_USE_C99 failed with the following output:
Change Dir: /data/centos/pytorch/build/CMakeFiles/CMakeTmp

Run Build Command(s):/data/centos/miniconda3/envs/simp/bin/ninja cmTC_9e0fb && [1/2] Building CXX object CMakeFiles/cmTC_9e0fb.dir/src.cxx.o
FAILED: CMakeFiles/cmTC_9e0fb.dir/src.cxx.o
/usr/bin/c++ -DSUPPORT_GLIBCXX_USE_C99  -std=c++14 -o CMakeFiles/cmTC_9e0fb.dir/src.cxx.o -c src.cxx
c++: error: unrecognized command line option '-std=c++14'
ninja: build stopped: subcommand failed.


Source file was:

  #include <cmath>
  #include <string>

  int main() {
    int a = std::isinf(3.0);
    int b = std::isnan(0.0);
    std::string s = std::to_string(1);

    return 0;
    }
Performing C++ SOURCE FILE Test SUPPORT_GLIBCXX_USE_C99 failed with the following output:
Change Dir: /data/centos/pytorch/build/CMakeFiles/CMakeTmp

Run Build Command(s):/data/centos/miniconda3/envs/simp/bin/ninja cmTC_aec96 && [1/2] Building CXX object CMakeFiles/cmTC_aec96.dir/src.cxx.o
FAILED: CMakeFiles/cmTC_aec96.dir/src.cxx.o
/usr/bin/c++ -DSUPPORT_GLIBCXX_USE_C99  -std=c++14 -o CMakeFiles/cmTC_aec96.dir/src.cxx.o -c src.cxx
c++: error: unrecognized command line option '-std=c++14'
ninja: build stopped: subcommand failed.


Source file was:

  #include <cmath>
  #include <string>

  int main() {
    int a = std::isinf(3.0);
    int b = std::isnan(0.0);
    std::string s = std::to_string(1);

    return 0;
    }
Performing C++ SOURCE FILE Test SUPPORT_GLIBCXX_USE_C99 failed with the following output:
Change Dir: /data/centos/pytorch/build/CMakeFiles/CMakeTmp

Run Build Command(s):/data/centos/miniconda3/envs/simp/bin/ninja cmTC_baf5b && [1/2] Building CXX object CMakeFiles/cmTC_baf5b.dir/src.cxx.o
FAILED: CMakeFiles/cmTC_baf5b.dir/src.cxx.o
/usr/bin/c++ -DSUPPORT_GLIBCXX_USE_C99  -std=c++14 -o CMakeFiles/cmTC_baf5b.dir/src.cxx.o -c src.cxx
c++: error: unrecognized command line option '-std=c++14'
ninja: build stopped: subcommand failed.


Source file was:

  #include <cmath>
  #include <string>

  int main() {
    int a = std::isinf(3.0);
    int b = std::isnan(0.0);
    std::string s = std::to_string(1);

    return 0;
    }