search

insilicomedicine / GENTRL

Generative Tensorial Reinforcement Learning (GENTRL) model
596 stars 215 forks source link

Keep get the error of torch.cuda.set_device(0) #36

Closed Nailo closed 1 year ago

Nailo commented 1 year ago

When I start the pretrain.ipynb, I got the AttributeError below. I have a rtx2060 on my laptop and cuda tool kit 11.4 installed.

AttributeError Traceback (most recent call last) Input In [10], in <cell line: 4>() 2 import torch 3 import pandas as pd ----> 4 torch.cuda.set_device(1)

File ~\miniconda3\envs\GENTRL\lib\site-packages\torch\cuda__init__.py:314, in set_device(device) 312 device = _get_device_index(device) 313 if device >= 0: --> 314 torch._C._cuda_setDevice(device)

AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'