search

NVIDIA / cub

[ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
BSD 3-Clause "New" or "Revised" License
1.68k stars 447 forks source link

Failed import cuda based pytorch extension #313

Closed LeopoldACC closed 3 years ago

LeopoldACC commented 3 years ago

When I try to reproduce the work,but the code stop at import cuda based extension as description in Issue link.

My environment is:

ubuntu20.04
python3.8.5
cuda11.1
cudnn8.0.4
nvidia driver460.73

torch1.8.1
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.73.01    Driver Version: 460.73.01    CUDA Version: 11.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  GeForce RTX 3090    Off  | 00000000:02:00.0  On |                  N/A |
|  0%   35C    P8    28W / 370W |    715MiB / 24265MiB |      3%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A       936      G   /usr/lib/xorg/Xorg                247MiB |
|    0   N/A  N/A      1325      G   /usr/bin/gnome-shell               58MiB |
|    0   N/A  N/A      1497      G   ...nlogin/bin/sunloginclient        6MiB |
|    0   N/A  N/A    487115      G   ...AAAAAAAA== --shared-files       46MiB |
|    0   N/A  N/A    595577      C   ...ffice/program/soffice.bin      253MiB |
|    0   N/A  N/A   1300314      G   ...AAAAAAAAA= --shared-files       96MiB |
+-----------------------------------------------------------------------------+
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Tue_Sep_15_19:10:02_PDT_2020
Cuda compilation tools, release 11.1, V11.1.74
Build cuda_11.1.TC455_06.29069683_0
alliepiper commented 3 years ago

CUB does not provide python bindings, and I don't see anything here that's actionable for our team.