search

ashawkey / stable-dreamfusion

Text-to-3D & Image-to-3D & Mesh Exportation with NeRF + Diffusion.
Apache License 2.0
7.99k stars 710 forks source link

['TORCH_CUDA_ARCH_LIST']. #360

Open Yhc-777 opened 1 month ago

Yhc-777 commented 1 month ago

Description

when i run :"python main.py -O --image ./df_ep0100_0008_rgb_rgba.png --workspace trial_image --iters 5000 --default_polar 80", i met this problem:“/home/u/miniconda3/envs/sdf/lib/python3.8/site-packages/torch/utils/cpp_extension.py:1967: UserWarning: TORCH_CUDA_ARCH_LIST is not set, all archs for visible cards are included for compilation. If this is not desired, please set os.environ['TORCH_CUDA_ARCH_LIST'].” How can I solve it? Thx!!

Steps to Reproduce

-

Expected Behavior

-

Environment

-

Duodecimus commented 1 day ago

The environment variable TORCH_CUDA_ARCH_LIST in PyTorch should be set to specify the CUDA architectures that you want PyTorch to build for when compiling CUDA kernels. Each architecture is specified by its CUDA Compute Capability version (such as 3.5, 5.2, 6.0, etc.).

You probably only want to list the value of the one graphics card you have. This is the same value as used for TCNN_CUDA_ARCHITECTURES or CUDA_ARCH

ypu can look your card up on https://developer.nvidia.com/cuda-gpus