Enviroment set up problem

[xuezhixie-cell]

Dear sir,

I tried to setup the env but keep getting problems. your help is really appreciated.

when using pip, i got error like below:

base) sagemaker-user@default:/gearnet_test/GearNet$ conda activate gearnet38_3
(gearnet38_3) sagemaker-user@default:/gearnet_test/GearNet$ pip install torch==1.8.0+cu111 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
Looking in links: https://download.pytorch.org/whl/lts/1.8/torch_lts.html
ERROR: Could not find a version that satisfies the requirement torch==1.8.0+cu111 (from versions: 1.4.0, 1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.8.1+cpu, 1.8.1+cu102, 1.8.1+cu111, 1.8.2+cpu, 1.8.2+cu101, 1.8.2+cu102, 1.8.2+cu111, 1.8.2+rocm3.10, 1.8.2+rocm4.0.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0, 1.12.0, 1.12.1, 1.13.0, 1.13.1, 2.0.0, 2.0.1, 2.1.0, 2.1.1, 2.1.2, 2.2.0, 2.2.1, 2.2.2, 2.3.0, 2.3.1, 2.4.0, 2.4.1)
ERROR: No matching distribution found for torch==1.8.0+cu111

when using conda, install part didnt show errors. but when run code. i got error for cuda.

(gearnet38_2) sagemaker-user@default:~/gearnet_test/GearNet$ python get_embedding.py
No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'
Traceback (most recent call last):
File "get_embedding.py", line 1, in
from torchdrug import models, data
File "/home/sagemaker-user/.conda/envs/gearnet38_2/lib/python3.8/site-packages/torchdrug/init.py", line 1, in
from . import patch
File "/home/sagemaker-user/.conda/envs/gearnet38_2/lib/python3.8/site-packages/torchdrug/patch.py", line 13, in
from torchdrug import core, data
File "/home/sagemaker-user/.conda/envs/gearnet38_2/lib/python3.8/site-packages/torchdrug/core/init.py", line 2, in
from .engine import Engine
File "/home/sagemaker-user/.conda/envs/gearnet38_2/lib/python3.8/site-packages/torchdrug/core/engine.py", line 11, in
from torchdrug import data, core, utils
File "/home/sagemaker-user/.conda/envs/gearnet38_2/lib/python3.8/site-packages/torchdrug/data/init.py", line 1, in
from .dictionary import PerfectHash, Dictionary
File "/home/sagemaker-user/.conda/envs/gearnet38_2/lib/python3.8/site-packages/torchdrug/data/dictionary.py", line 4, in
from torch_scatter import scatter_max
File "/home/sagemaker-user/.conda/envs/gearnet38_2/lib/python3.8/site-packages/torch_scatter/init.py", line 16, in
torch.ops.load_library(spec.origin)
File "/home/sagemaker-user/.conda/envs/gearnet38_2/lib/python3.8/site-packages/torch/_ops.py", line 104, in load_library
ctypes.CDLL(path)
File "/home/sagemaker-user/.conda/envs/gearnet38_2/lib/python3.8/ctypes/init.py", line 373, in init
self._handle = _dlopen(self._name, mode)
OSError: libc10_cuda.so: cannot open shared object file: No such file or directory

Thanks for your help.