ArtEq
ArtEq copied to clipboard
Published
20 hours ago •
HavenFeng
HavenFeng
ModuleNotFoundError: No module named 'epn_grouping'
Failed to runt training. Here is the error
Traceback (most recent call last):
File "/home/blinkdrive/Documents/UNT/Project/Spring 2024/02ArtEq/src/eval.py", line 12, in <module>
from models_pointcloud import PointCloud_network_equiv
File "/home/blinkdrive/Documents/UNT/Project/Spring 2024/02ArtEq/src/models_pointcloud.py", line 9, in <module>
from so3conv import so3_mean
File "/home/blinkdrive/Documents/UNT/Project/Spring 2024/02ArtEq/src/so3conv.py", line 4, in <module>
import vgtk.so3conv as sptk
File "/home/blinkdrive/anaconda3/envs/arteq/lib/python3.10/site-packages/vgtk/__init__.py", line 3, in <module>
from . import pc
File "/home/blinkdrive/anaconda3/envs/arteq/lib/python3.10/site-packages/vgtk/pc/__init__.py", line 4, in <module>
from .sample import *
File "/home/blinkdrive/anaconda3/envs/arteq/lib/python3.10/site-packages/vgtk/pc/sample.py", line 5, in <module>
import epn_grouping as cuda_nn
ModuleNotFoundError: No module named 'epn_grouping
my conda installation
# packages in environment at /home/blinkdrive/anaconda3/envs/arteq:
#
# Name Version Build Channel
_libgcc_mutex 0.1 conda_forge conda-forge
_openmp_mutex 4.5 2_gnu conda-forge
addict 2.4.0 pypi_0 pypi
asttokens 2.4.1 pypi_0 pypi
attrs 23.2.0 pypi_0 pypi
blas 1.0 mkl conda-forge
blinker 1.7.0 pypi_0 pypi
braceexpand 0.1.7 pypi_0 pypi
brotli-python 1.0.9 py310h6a678d5_7
bzip2 1.0.8 hd590300_5 conda-forge
c-ares 1.28.1 hd590300_0 conda-forge
ca-certificates 2024.3.11 h06a4308_0
ccimport 0.4.2 pypi_0 pypi
certifi 2024.2.2 pyhd8ed1ab_0 conda-forge
cffi 1.16.0 py310h2fee648_0 conda-forge
cfgv 3.3.1 pyhd8ed1ab_0 conda-forge
charset-normalizer 2.0.4 pyhd3eb1b0_0
click 8.1.7 pypi_0 pypi
colorama 0.4.6 pyhd8ed1ab_0 conda-forge
comm 0.2.2 pypi_0 pypi
configargparse 1.7 pypi_0 pypi
contourpy 1.2.0 pypi_0 pypi
cpuonly 2.0 0 pytorch
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
cumm 0.4.11 pypi_0 pypi
curl 8.7.1 hca28451_0 conda-forge
cycler 0.12.1 pypi_0 pypi
dash 2.16.1 pypi_0 pypi
dash-core-components 2.0.0 pypi_0 pypi
dash-html-components 2.0.0 pypi_0 pypi
dash-table 5.0.0 pypi_0 pypi
decorator 5.1.1 pypi_0 pypi
distlib 0.3.8 pyhd8ed1ab_0 conda-forge
exceptiongroup 1.2.0 pypi_0 pypi
executing 2.0.1 pypi_0 pypi
fastjsonschema 2.19.1 pypi_0 pypi
ffmpeg 4.3 hf484d3e_0 pytorch
filelock 3.13.3 pyhd8ed1ab_0 conda-forge
fire 0.6.0 pypi_0 pypi
flask 3.0.2 pypi_0 pypi
fonttools 4.50.0 pypi_0 pypi
freetype 2.12.1 h4a9f257_0
fsspec 2024.3.1 pyhca7485f_0 conda-forge
gettext 0.21.1 h27087fc_0 conda-forge
git 2.44.0 pl5321h709897a_0 conda-forge
gmp 6.3.0 h59595ed_1 conda-forge
gmpy2 2.1.2 py310h3ec546c_1 conda-forge
gnutls 3.6.15 he1e5248_0
huggingface-hub 0.22.2 pypi_0 pypi
identify 2.5.35 pyhd8ed1ab_0 conda-forge
idna 3.4 py310h06a4308_0
importlib-metadata 7.1.0 pypi_0 pypi
intel-openmp 2022.1.0 h9e868ea_3769
ipython 8.23.0 pypi_0 pypi
ipywidgets 8.1.2 pypi_0 pypi
itsdangerous 2.1.2 pypi_0 pypi
jedi 0.19.1 pypi_0 pypi
jinja2 3.1.3 pyhd8ed1ab_0 conda-forge
joblib 1.3.2 pyhd8ed1ab_0 conda-forge
jpeg 9e h5eee18b_1
jsonschema 4.21.1 pypi_0 pypi
jsonschema-specifications 2023.12.1 pypi_0 pypi
jupyter-core 5.7.2 pypi_0 pypi
jupyterlab-widgets 3.0.10 pypi_0 pypi
keyutils 1.6.1 h166bdaf_0 conda-forge
kiwisolver 1.4.5 pypi_0 pypi
krb5 1.21.2 h659d440_0 conda-forge
lame 3.100 h7b6447c_0
lark 1.1.9 pypi_0 pypi
lcms2 2.12 h3be6417_0
ld_impl_linux-64 2.40 h41732ed_0 conda-forge
lerc 3.0 h295c915_0
libblas 3.9.0 16_linux64_mkl conda-forge
libcblas 3.9.0 16_linux64_mkl conda-forge
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
libcurl 8.7.1 hca28451_0 conda-forge
libcusolver 11.4.1.48 0 nvidia
libcusparse 11.7.5.86 0 nvidia
libdeflate 1.8 h7f8727e_5
libedit 3.1.20191231 he28a2e2_2 conda-forge
libev 4.33 hd590300_2 conda-forge
libexpat 2.6.2 h59595ed_0 conda-forge
libffi 3.4.2 h7f98852_5 conda-forge
libgcc-ng 13.2.0 h807b86a_5 conda-forge
libgfortran-ng 13.2.0 h69a702a_5 conda-forge
libgfortran5 13.2.0 ha4646dd_5 conda-forge
libgomp 13.2.0 h807b86a_5 conda-forge
libiconv 1.17 hd590300_2 conda-forge
libidn2 2.3.4 h5eee18b_0
libjpeg-turbo 2.0.0 h9bf148f_0 pytorch
liblapack 3.9.0 16_linux64_mkl conda-forge
libnghttp2 1.58.0 h47da74e_1 conda-forge
libnpp 11.8.0.86 0 nvidia
libnsl 2.0.1 hd590300_0 conda-forge
libnvjpeg 11.9.0.86 0 nvidia
libpng 1.6.39 h5eee18b_0
libsqlite 3.45.2 h2797004_0 conda-forge
libssh2 1.11.0 h0841786_0 conda-forge
libstdcxx-ng 13.2.0 h7e041cc_5 conda-forge
libtasn1 4.19.0 h5eee18b_0
libtiff 4.4.0 hecacb30_2
libunistring 0.9.10 h27cfd23_0
libuuid 2.38.1 h0b41bf4_0 conda-forge
libwebp-base 1.3.2 h5eee18b_0
libxcrypt 4.4.36 hd590300_1 conda-forge
libzlib 1.2.13 hd590300_5 conda-forge
llvm-openmp 15.0.7 h0cdce71_0 conda-forge
loguru 0.7.2 pypi_0 pypi
markupsafe 2.1.5 py310h2372a71_0 conda-forge
matplotlib 3.8.3 pypi_0 pypi
matplotlib-inline 0.1.6 pypi_0 pypi
mkl 2022.1.0 hc2b9512_224
mpc 1.3.1 hfe3b2da_0 conda-forge
mpfr 4.2.1 h9458935_0 conda-forge
mpmath 1.3.0 pyhd8ed1ab_0 conda-forge
nbformat 5.10.3 pypi_0 pypi
ncurses 6.4.20240210 h59595ed_0 conda-forge
nest-asyncio 1.6.0 pypi_0 pypi
nettle 3.7.3 hbbd107a_1
networkx 3.2.1 pyhd8ed1ab_0 conda-forge
ninja 1.11.1.1 pypi_0 pypi
nodeenv 1.8.0 pyhd8ed1ab_0 conda-forge
numpy 1.26.4 py310hb13e2d6_0 conda-forge
nvidia-cublas-cu11 11.11.3.6 pypi_0 pypi
nvidia-cuda-cupti-cu11 11.8.87 pypi_0 pypi
nvidia-cuda-nvrtc-cu11 11.8.89 pypi_0 pypi
nvidia-cuda-runtime-cu11 11.8.89 pypi_0 pypi
nvidia-cudnn-cu11 8.7.0.84 pypi_0 pypi
nvidia-cufft-cu11 10.9.0.58 pypi_0 pypi
nvidia-curand-cu11 10.3.0.86 pypi_0 pypi
nvidia-cusolver-cu11 11.4.1.48 pypi_0 pypi
nvidia-cusparse-cu11 11.7.5.86 pypi_0 pypi
nvidia-nccl-cu11 2.19.3 pypi_0 pypi
nvidia-nvtx-cu11 11.8.86 pypi_0 pypi
open3d 0.18.0 pypi_0 pypi
openh264 2.1.1 h4ff587b_0
openjpeg 2.4.0 h3ad879b_0
openssl 3.2.1 hd590300_1 conda-forge
packaging 24.0 pypi_0 pypi
pandas 2.2.1 pypi_0 pypi
parso 0.8.3 pypi_0 pypi
pccm 0.4.11 pypi_0 pypi
pcre2 10.43 hcad00b1_0 conda-forge
perl 5.32.1 7_hd590300_perl5 conda-forge
pexpect 4.9.0 pypi_0 pypi
pillow 10.2.0 py310h5eee18b_0
pip 24.0 pyhd8ed1ab_0 conda-forge
platformdirs 4.2.0 pyhd8ed1ab_0 conda-forge
plotly 5.20.0 pypi_0 pypi
plyfile 1.0.3 pyhd8ed1ab_0 conda-forge
portalocker 2.8.2 pypi_0 pypi
pre-commit 3.7.0 pyha770c72_0 conda-forge
prompt-toolkit 3.0.43 pypi_0 pypi
ptyprocess 0.7.0 pypi_0 pypi
pure-eval 0.2.2 pypi_0 pypi
pybind11 2.12.0 pypi_0 pypi
pycparser 2.22 pyhd8ed1ab_0 conda-forge
pygments 2.17.2 pypi_0 pypi
pyparsing 3.1.2 pypi_0 pypi
pyquaternion 0.9.9 pypi_0 pypi
pysocks 1.7.1 py310h06a4308_0
python 3.10.14 hd12c33a_0_cpython conda-forge
python-dateutil 2.9.0.post0 pypi_0 pypi
python_abi 3.10 4_cp310 conda-forge
pytorch-cuda 11.8 h7e8668a_5 pytorch
pytorch-mutex 1.0 cpu pytorch
pytorch-scatter 2.1.2 py310_torch_2.2.0_cu118 pyg
pytz 2024.1 pypi_0 pypi
pyyaml 6.0.1 py310h2372a71_1 conda-forge
readline 8.2 h8228510_1 conda-forge
referencing 0.34.0 pypi_0 pypi
requests 2.31.0 py310h06a4308_1
retrying 1.3.4 pypi_0 pypi
rpds-py 0.18.0 pypi_0 pypi
scikit-learn 1.4.1.post1 py310h1fdf081_0 conda-forge
scipy 1.12.0 py310hb13e2d6_2 conda-forge
setuptools 69.2.0 pyhd8ed1ab_0 conda-forge
six 1.16.0 pypi_0 pypi
smplx 0.1.28 pypi_0 pypi
spconv 2.3.6 pypi_0 pypi
stack-data 0.6.3 pypi_0 pypi
sympy 1.12 pypyh9d50eac_103 conda-forge
tenacity 8.2.3 pypi_0 pypi
termcolor 2.4.0 pypi_0 pypi
threadpoolctl 3.4.0 pyhc1e730c_0 conda-forge
tk 8.6.13 noxft_h4845f30_101 conda-forge
torch 2.2.1+cu118 pypi_0 pypi
torchaudio 2.2.1+cu118 pypi_0 pypi
torchvision 0.17.1+cu118 pypi_0 pypi
tqdm 4.66.2 pyhd8ed1ab_0 conda-forge
traitlets 5.14.2 pypi_0 pypi
trimesh 4.2.4 pyhd8ed1ab_0 conda-forge
triton 2.2.0 pypi_0 pypi
typing_extensions 4.10.0 pyha770c72_0 conda-forge
tzdata 2024.1 pypi_0 pypi
ukkonen 1.0.1 py310hd41b1e2_4 conda-forge
urllib3 2.1.0 py310h06a4308_1
vgtk 1.0 pypi_0 pypi
virtualenv 20.25.1 pyhd8ed1ab_0 conda-forge
wcwidth 0.2.13 pypi_0 pypi
webdataset 0.2.86 pypi_0 pypi
werkzeug 3.0.2 pypi_0 pypi
wheel 0.43.0 pyhd8ed1ab_1 conda-forge
widgetsnbextension 4.0.10 pypi_0 pypi
xz 5.2.6 h166bdaf_0 conda-forge
yacs 0.1.8 pypi_0 pypi
yaml 0.2.5 h7f98852_2 conda-forge
zipp 3.18.1 pypi_0 pypi
zlib 1.2.13 hd590300_5 conda-forge
zstd 1.5.5 hfc55251_0 conda-forge
my GPU and driver
Mon Apr 1 21:58:07 2024
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 525.147.05 Driver Version: 525.147.05 CUDA Version: 12.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce ... On | 00000000:09:00.0 On | N/A |
| 0% 49C P8 39W / 370W | 558MiB / 24576MiB | 15% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
Pytorch installed
~/Documents/UNT/Project/Spring 2024/02ArtEq$ python -c "import torch; print('PyTorch version:', torch.__version__); print('CUDA available:', torch.cuda.is_available()); print('CUDA version:', torch.version.cuda)"
PyTorch version: 2.2.1+cu118
CUDA available: True
CUDA version: 11.8
