SMOKE
SMOKE copied to clipboard
Run SMOKE on cuda 11.x environment!
I struggled to run SMOKE model on cuda 11.x environment for last 4 months and finally found a simple solution.
now you can run SMOKE model on cuda 11.x environment.
This methods tested on following environment
- gpu : quadro rtx 6000
- driver version : 460.73.01
- cuda version : 11.0
- pytorch : 1.8.0
In future I'm planning to test on geforce RTX 3060. If someone tested on RTX 30's series, share us results!
Hey, I followed your solution and got the following error on RTX 3090:
subprocess.CalledProcessError: Command '['which', 'g++']' returned non-zero exit status 1.
GPU: RTX 3090 Driver version: 510.06 cuda version: 11.4 pytorch: 1.8.1
Entire error message:
Hey, I followed your solution and got the following error on RTX 3090:
subprocess.CalledProcessError: Command '['which', 'g++']' returned non-zero exit status 1.
GPU: RTX 3090 Driver version: 510.06 cuda version: 11.4 pytorch: 1.8.1
Entire error message:
Oh nvm. I'm using docker and I forget to install gcc
Worked for me in google colab
running build
running build_ext
/home/awguitar/.local/lib/python3.8/site-packages/torch/utils/cpp_extension.py:387: UserWarning: Attempted to use ninja as the BuildExtension backend but we could not find ninja.. Falling back to using the slow distutils backend.
warnings.warn(msg.format('we could not find ninja.'))
Traceback (most recent call last):
File "setup.py", line 61, in
Hi, @liukr001 I think pytorch version 10.2 mismatches with with local CUDA version 11.4. I found similar issue. Please refer to this link. https://github.com/rusty1s/pytorch_sparse/issues/189#issuecomment-1000240508
@yhchang1120 What results did you get in your environment?
Worked for me in google colab
Can you share your note book in ggloab with me,pls!