RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`

オープン
#369 コメント 2 件 リアクション 0 件 担当者 0 名 GitHub で見る

まだ誰も着手していません。

評価

難易度
4/5
見積もり時間
3〜5日
初心者へのやさしさ
25/100
issue の種類
バグ
明瞭さ
説明が足りない
活発さ
停滞
技術スタック
python, pytorch

調査の方向性

Reproduce the macrocyclic example through scripts/run_inference.py using the listed Hydra overrides. Trace the failing call through rfdiffusion/inference/model_runners.py, RoseTTAFoldModel.py, and Embeddings.py, and inspect the PyTorch CUDA environment around the cublasCreate failure. Done means the example completes inference without the CUBLAS_STATUS_NOT_INITIALIZED error.

索引モデルが issue の本文から書いたものです。

説明

Hi community,
While running the macrocyclic example, I got this issue:

Error executing job with overrides: ['inference.output_prefix=my_files/out_macrocycle/macrocyclic_test', 'contigmap.contigs=[10-18]', 'inference.cyclic=True', "inference.cyc_chains='a'", 'inference.num_designs=5', 'diffuser.T=50']
Traceback (most recent call last):
File "D:\Peptide Design\RFdiffusion\scripts\run_inference.py", line 94, in main
px0, x_t, seq_t, plddt = sampler.sample_step(
File "d:\peptide design\rfdiffusion\rfdiffusion\inference\model_runners.py", line 686, in sample_step
msa_prev, pair_prev, px0, state_prev, alpha, logits, plddt = self.model(msa_masked,
File "D:\miniconda3\envs\SE3nv\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "d:\peptide design\rfdiffusion\rfdiffusion\RoseTTAFoldModel.py", line 77, in forward
msa_latent, pair, state = self.latent_emb(msa_latent, seq, idx, cyclic_reses)
File "D:\miniconda3\envs\SE3nv\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "d:\peptide design\rfdiffusion\rfdiffusion\Embeddings.py", line 96, in forward
msa = self.emb(msa) # (B, N, L, d_model) # MSA embedding
File "D:\miniconda3\envs\SE3nv\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "D:\miniconda3\envs\SE3nv\lib\site-packages\torch\nn\modules\linear.py", line 96, in forward
return F.linear(input, self.weight, self.bias)
File "D:\miniconda3\envs\SE3nv\lib\site-packages\torch\nn\functional.py", line 1847, in linear
return torch._C._nn.linear(input, weight, bias)
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle)

Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.

Nobody got this so far. Please help me solve this problem

主要言語
Python
スター
3.1k
フォーク
644
PR マージ指標
30日以内にマージされた PR はありません

コントリビューションガイド

このリポジトリのコントリビューションガイドは索引されていません

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

RosettaCommons/RFdiffusion のほかの issue

RosettaCommons/RFdiffusion の issue をすべて見る

似ている issue

Python の issue をもっと見る

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。