search

nagadomi / distro

Unofficial maintenance repository of Torch7. It supports CUDA10.1, Volta, Turing, Docker https://hub.docker.com/r/nagadomi/torch7
BSD 3-Clause "New" or "Revised" License
201 stars 57 forks source link

cuda version 10.2 #18

Open arjunsankar opened 1 year ago

arjunsankar commented 1 year ago

Hi @nagadomi , I got this error.

-- Autodetected CUDA architecture(s): 8.6 -- got cuda version 10.2

nvcc fatal : Unsupported gpu architecture 'compute_86' nvcc fatal : Unsupported gpu architecture 'compute_86'

nagadomi commented 1 year ago

For unsupported gpu device in CUDA 10, Try building with the following command.

TORCH_CUDA_ARCH_LIST="7.5+PTX" ./install.sh

Then, the first time the application is run, it is JIT compiled into the device binary with NVIDIA Driver. This will take several minutes.

nagadomi commented 1 year ago

I pushed a Docker Image with the above changes, see README about JIT Caching. https://hub.docker.com/r/nagadomi/torch7