adli-igc comments

Repositories
Issues
Comments

Results 2 comments of


                                            adli-igc

CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' Building PyTorch extension for tiny-cuda-nn version 1.7

You have to specify the TCNN_CUDA_ARCHITECTURE as shown in this line of code: ``` if "TCNN_CUDA_ARCHITECTURES" in os.environ and os.environ["TCNN_CUDA_ARCHITECTURES"]: compute_capabilities = [int(x) for x in os.environ["TCNN_CUDA_ARCHITECTURES"].replace(";", ",").split(",")] print(f"Obtained compute...

CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' Building PyTorch extension for tiny-cuda-nn version 1.7

This is an example of my Dockerfile where i've set the CUDA architecture ``` ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:22.05-py3 FROM $BASE_IMAGE RUN apt-get update -yq --fix-missing \ && DEBIAN_FRONTEND=noninteractive apt-get install -yq --no-install-recommends...