mhamilton723 / FeatUp

Official code for "FeatUp: A Model-Agnostic Frameworkfor Features at Any Resolution" ICLR 2024

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Pip install error

JaYzZ opened this issue · comments

featup_err.txt

Any idea on why this error attached is happening during pip install? Here are the key conda environment components:

Name Version Build Channel
_libgcc_mutex 0.1 main defaults
_openmp_mutex 5.1 1_gnu defaults
blas 1.0 mkl defaults
clip 1.0 pypi_0 pypi
clip-benchmark 1.6.1 pypi_0 pypi
cuda-cudart 11.8.89 0 nvidia
cuda-cupti 11.8.87 0 nvidia
cuda-libraries 11.8.0 0 nvidia
cuda-nvrtc 11.8.89 0 nvidia
cuda-nvtx 11.8.86 0 nvidia
cuda-runtime 11.8.0 0 nvidia
intel-openmp 2023.1.0 hdb19cb5_46306 defaults
kornia 0.7.2 pypi_0 pypi
kornia-rs 0.1.3 pypi_0 pypi
libcublas 11.11.3.6 0 nvidia
libcufft 10.9.0.58 0 nvidia
libcufile 1.9.0.20 0 nvidia
libcurand 10.3.5.119 0 nvidia
libcusolver 11.4.1.48 0 nvidia
libcusparse 11.7.5.86 0 nvidia
llvm-openmp 14.0.6 h9e868ea_0 defaults
mkl 2023.1.0 h213fc3f_46344 defaults
mkl-service 2.4.0 py39h5eee18b_1 defaults
mkl_fft 1.3.8 py39h5eee18b_0 defaults
mkl_random 1.2.4 py39hdb19cb5_0 defaults
ninja 1.11.1.1 pypi_0 pypi
nvidia-ml-py 12.535.133 pypi_0 pypi
pip 23.3.1 py39h06a4308_0 defaults
python 3.9.18 h955ad1f_0 defaults
pytorch 2.2.2 py3.9_cuda11.8_cudnn8.7.0_0 pytorch
pytorch-cuda 11.8 h7e8668a_5 pytorch
pytorch-lightning 2.1.3 pypi_0 pypi
pytorch-mutex 1.0 cuda pytorch
setuptools 68.0.0 py39h06a4308_0 defaults
tensorboard 2.15.1 pypi_0 pypi
tensorboard-data-server 0.7.2 pypi_0 pypi
timm 0.9.16 pypi_0 pypi
torch-kmeans 0.2.0 pypi_0 pypi
torchmetrics 1.2.1 pypi_0 pypi
torchtriton 2.2.0 py39 pytorch
torchvision 0.17.2 py39_cu118 pytorch
wheel 0.41.2 py39h06a4308_0 defaults
xformers 0.0.25.post1 py39_cu11.8.0_pyt2.2.2 xformers

You've installed pytorch compiled with CUDA 11.8 when your machine is running CUDA v11.1

Try updating CUDA on your machine to 11.8