Skip to content

CUDA error #71

Description

@coder1165

When run code " bash ./scripts/MSL.sh" raised error:
Traceback (most recent call last):
File "main.py", line 52, in
main(config)
File "main.py", line 18, in main
solver = Solver(vars(config))
File "/storage/student1/c_cxl/cxl/Anomaly-Transformer/solver.py", line 85, in init
self.build_model()
File "/storage/student1/c_cxl/cxl/Anomaly-Transformer/solver.py", line 90, in build_model
self.model = AnomalyTransformer(win_size=self.win_size, enc_in=self.input_c, c_out=self.output_c, e_layers=3)
File "/storage/student1/c_cxl/cxl/Anomaly-Transformer/model/AnomalyTransformer.py", line 77, in init
) for l in range(e_layers)
File "/storage/student1/c_cxl/cxl/Anomaly-Transformer/model/AnomalyTransformer.py", line 77, in
) for l in range(e_layers)
File "/storage/student1/c_cxl/cxl/Anomaly-Transformer/model/attn.py", line 29, in init
self.distances = torch.zeros((window_size, window_size)).cuda()
File "/storage/student1/c_cxl/miniconda3/envs/Anomaly_Trans/lib/python3.6/site-packages/torch/cuda/init.py", line 214, in _lazy_init
torch._C._cuda_init()
RuntimeError: No CUDA GPUs are available

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions