Open arjunsankar opened 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.
I pushed a Docker Image with the above changes, see README about JIT Caching. https://hub.docker.com/r/nagadomi/torch7
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'