You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Make sure the Anaconda has been installed in your computer. All the experiments are conducted on a Linux Server equipped with NVIDIA GeForce RTX 3090 24GB. Make sure your GPU environment has enough memory. CPU is not supported. CUDA120 and above is not supported.
(1) Run the following command
conda env create -f deepinfer.yml
conda activate deepinfer
Activate this environment and install the following libraries.
(1) torch==1.11.0+cu113
pip install torch-1.11.0+cu113-cp39-cp39-linux_x86_64.whl Please download the whl file from <https://download.pytorch.org/whl/torch/>
(2) torch-scatter==2.0.9 and torch-sparse==0.6.15
pip install torch_scatter-2.0.9-cp39-cp39-linux_x86_64.whl
pip install torch_sparse-0.6.15-cp39-cp39-linux_x86_64.whl
Please download the whl file from <https://pytorch-geometric.com/whl/torch-1.11.0%2Bcu113.html>
(3) torch-geometric==2.1.0.post1
pip install torch-geometric==2.1.0.post1
Note: Please update the CUDA version according to your actual version for pytorch.