Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc) #77

Open
LLL-YUE opened this issue Jun 21, 2024 · 0 comments

Comments

@LLL-YUE
Copy link

LLL-YUE commented Jun 21, 2024

When I run experiment.py,the mistakes appear as follows:
NVIDIA GeForce RTX 3060 Laptop GPU with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA GeForce RTX 3060 Laptop GPU GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
Traceback (most recent call last):
File "/home/liuyue/projects/decision-transformer-master/gym/experiment.py", line 308, in
experiment('gym-experiment', variant=vars(args))
File "/home/liuyue/projects/decision-transformer-master/gym/experiment.py", line 278, in experiment
outputs = trainer.train_iteration(num_steps=variant['num_steps_per_iter'], iter_num=iter+1, print_logs=True)
File "/home/liuyue/projects/decision-transformer-master/gym/decision_transformer/training/trainer.py", line 30, in train_iteration
train_loss = self.train_step()
File "/home/liuyue/projects/decision-transformer-master/gym/decision_transformer/training/seq_trainer.py", line 13, in train_step
state_preds, action_preds, reward_preds = self.model.forward(
File "/home/liuyue/projects/decision-transformer-master/gym/decision_transformer/models/decision_transformer.py", line 63, in forward
state_embeddings = self.embed_state(states)
File "/home/liuyue/anaconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/liuyue/anaconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/torch/nn/modules/linear.py", line 94, in forward
return F.linear(input, self.weight, self.bias)
File "/home/liuyue/anaconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/torch/nn/functional.py", line 1753, in linear
return torch._C._nn.linear(input, weight, bias)
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant