编译torch记录

环境

cuda drvier 11.6
cuda toolkit 11.1
pytorch ver 1.11

conda env

# conda package list
# packages in environment at /home/tangke/anaconda3/envs/py39torch:
#
# Name                    Version                   Build  Channel
_libgcc_mutex             0.1                        main  
_openmp_mutex             5.1                       1_gnu  
astunparse                1.6.3                      py_0  
blas                      1.0                         mkl  
brotlipy                  0.7.0           py39h27cfd23_1003  
bzip2                     1.0.8                h7b6447c_0  
c-ares                    1.18.1               h7f8727e_0  
ca-certificates           2022.4.26            h06a4308_0  
ccache                    4.5.1                haef5404_0    conda-forge
certifi                   2022.5.18.1      py39h06a4308_0  
cffi                      1.15.0           py39hd667e15_1  
charset-normalizer        2.0.4              pyhd3eb1b0_0  
cmake                     3.22.1               h1fce559_0  
cryptography              37.0.1           py39h9ce1e76_0  
dataclasses               0.6                      pypi_0    pypi
expat                     2.4.4                h295c915_0  
expecttest                0.1.3              pyhd3eb1b0_0  
future                    0.18.2           py39h06a4308_1  
idna                      3.3                pyhd3eb1b0_0  
intel-openmp              2021.4.0          h06a4308_3561  
krb5                      1.19.2               hac12032_0  
ld_impl_linux-64          2.38                 h1181459_1  
libcurl                   7.82.0               h0b77cf5_0  
libedit                   3.1.20210910         h7f8727e_0  
libev                     4.33                 h7f8727e_1  
libffi                    3.3                  he6710b0_2  
libgcc-ng                 11.2.0               h1234567_1  
libgfortran-ng            11.2.0               h00389a5_1  
libgfortran5              11.2.0               h1234567_1  
libgomp                   11.2.0               h1234567_1  
libhiredis                1.0.2                h2cc385e_0    conda-forge
libnghttp2                1.46.0               hce63b2e_0  
libssh2                   1.10.0               h8f2d780_0  
libstdcxx-ng              11.2.0               h1234567_1  
libuv                     1.40.0               h7b6447c_0  
lz4-c                     1.9.3                h295c915_1  
magma-cuda111             2.5.2                         1    pytorch
mkl                       2021.4.0           h06a4308_640  
mkl-include               2022.0.1           h06a4308_117  
mkl-service               2.4.0            py39h7f8727e_0  
mkl_fft                   1.3.1            py39hd3c417c_0  
mkl_random                1.2.2            py39h51133e4_0  
ncurses                   6.3                  h7f8727e_2  
ninja                     1.10.2               h06a4308_5  
ninja-base                1.10.2               hd09550d_5  
numpy                     1.22.3           py39he7a7128_0  
numpy-base                1.22.3           py39hf524024_0  
openssl                   1.1.1o               h7f8727e_0  
pip                       21.2.4           py39h06a4308_0  
psutil                    5.8.0            py39h27cfd23_1  
pycparser                 2.21               pyhd3eb1b0_0  
pyopenssl                 22.0.0             pyhd3eb1b0_0  
pysocks                   1.7.1            py39h06a4308_0  
python                    3.9.12               h12debd9_0  
pyyaml                    6.0              py39h7f8727e_1  
readline                  8.1.2                h7f8727e_1  
requests                  2.27.1             pyhd3eb1b0_0  
rhash                     1.4.1                h3c74f83_1  
setuptools                59.5.0                   pypi_0    pypi
six                       1.16.0             pyhd3eb1b0_1  
sqlite                    3.38.3               hc218d9a_0  
tk                        8.6.12               h1ccaba5_0  
torch                     1.11.0a0+git8d365ae           dev_0    <develop>
types-dataclasses         0.6.5                    pypi_0    pypi
typing_extensions         4.1.1              pyh06a4308_0  
tzdata                    2022a                hda174b7_0  
urllib3                   1.26.9           py39h06a4308_0  
wheel                     0.37.1             pyhd3eb1b0_0  
xz                        5.2.5                h7f8727e_1  
yaml                      0.2.5                h7b6447c_0  
zlib                      1.2.12               h7f8727e_2  
zstd                      1.5.2                ha4553b6_0  

# pip package list
Package            Version             Location
------------------ ------------------- --------------------
astunparse         1.6.3
brotlipy           0.7.0
certifi            2022.5.18.1
cffi               1.15.0
charset-normalizer 2.0.4
cryptography       37.0.1
dataclasses        0.6
expecttest         0.1.3
future             0.18.2
idna               3.3
mkl-fft            1.3.1
mkl-random         1.2.2
mkl-service        2.4.0
ninja              1.10.2.3
numpy              1.22.3
pip                21.2.4
psutil             5.8.0
pycparser          2.21
pyOpenSSL          22.0.0
PySocks            1.7.1
PyYAML             6.0
requests           2.27.1
setuptools         59.5.0
six                1.16.0
torch              1.11.0a0+git8d365ae /home/tangke/pytorch
types-dataclasses  0.6.5
typing_extensions  4.1.1
urllib3            1.26.9
wheel              0.37.1

编译命令
USE_CUDA=1 DEBUG=1 CUDA_VISIBLE_DEVICES=4 python3 setup.py develop