ZauraizAlamgeer
commented
1 year ago Hey I got the same error and here is how I fixed it. Use Conda environment. Please remove your old environment and create new as follows and you should be good:
Create new environment
Give it a name and please use python3.8 and no later since newer version is not supported by a few dependencies.
$ conda create -n
Install this specific PyTorch and CUDA toolkit version
$ conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
Then Install other Dependencies
$ pip install wandb $ pip install --ignore-installed PyYAML $ pip install open3d $ pip install multimethod $ pip install termcolor $ pip install trimesh $ pip install easydict
To remove the error related to libcublas.
$ conda install -c "nvidia/label/cuda-11.7.0" cuda
Then install pointops
$ cd external_libs/pointops & python setup.py install
Now you are good to. :)
Brin333
coordxyz
Frq-F
ericnlt
1404561326521
SunnyRT
Under the 'ToothGroupNetwork-challenge_branch' fold, exec the code: python inference_final.py --input_path ./tmp --save_path ./results
Error:: Error(s) in loading state_dict for TfCblFirstModule: While copying the parameter named "first_ins_cent_model.enc1.0.linear.weight", whose dimensions in the model are torch.Size([32, 6]) and whose dimensions in the checkpoint are torch.Size([32, 6]), an exception occurred : ('CUDA error: no kernel image is available for execution on the device',).