質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
86.12%
CUDA

CUDAは並列計算プラットフォームであり、Nvidia GPU(Graphics Processing Units)向けのプログラミングモデルです。CUDAは様々なプログラミング言語、ライブラリ、APIを通してNvidiaにインターフェイスを提供します。

PyTorch

PyTorchは、オープンソースのPython向けの機械学習ライブラリ。Facebookの人工知能研究グループが開発を主導しています。強力なGPUサポートを備えたテンソル計算、テープベースの自動微分による柔軟なニューラルネットワークの記述が可能です。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

解決済

RTX3090+cuda10.2の環境でPyTorchが動作しない

YuYuMoYuYu
YuYuMoYuYu

総合スコア8

CUDA

CUDAは並列計算プラットフォームであり、Nvidia GPU(Graphics Processing Units)向けのプログラミングモデルです。CUDAは様々なプログラミング言語、ライブラリ、APIを通してNvidiaにインターフェイスを提供します。

PyTorch

PyTorchは、オープンソースのPython向けの機械学習ライブラリ。Facebookの人工知能研究グループが開発を主導しています。強力なGPUサポートを備えたテンソル計算、テープベースの自動微分による柔軟なニューラルネットワークの記述が可能です。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

1回答

0グッド

0クリップ

9036閲覧

投稿2021/09/09 07:51

編集2021/09/09 08:14

前提・実現したいこと

RTX3090+cuda10.2の環境でPyTorch(torch 1.9.0)を使ったディープラーニングを実装しようとしています。

発生している問題・エラーメッセージ

C:\Users\user-name\anaconda3\envs\torch\lib\site-packages\torch\cuda\__init__.py:106: UserWarning: GeForce RTX 3090 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_61 sm_70 sm_75 compute_37. If you want to use the GeForce RTX 3090 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))```

該当のソースコード

python

1import torch 2 3print(torch.tensor([0.1, 0.2], device='cpu')) 4# tensor([0.1000, 0.2000]) 5 6print(torch.tensor([0.1, 0.2], device='cuda')) 7# 上記エラーが発生

試したこと

PyTorchがそもそも対応しているかどうかを確認しました。

python

1import torch 2print(torch.cuda.get_arch_list()) 3# ['sm_37', 'sm_50', 'sm_60', 'sm_61', 'sm_70', 'sm_75', 'compute_37']

このリストにはRTX3090のsm_86がないことが確認できます。
が,調べていくと,ソースコードを3行程度修正すればパスするとの報告があります。
https://www.gdep.co.jp/information/hot/geforce-rtx3090-3080-3070%e3%81%ab%e3%81%a4%e3%81%84%e3%81%a6/

ただ,どこをどう修正すればよいか検討がつかず。。。
お詳しい方,ご教示いただけないでしょうか。。。

補足情報(FW/ツールのバージョンなど)

Windows 10 Pro
Intel(R) Core(TM) i9-10980XE CPU @3.00GHz
NVIDIA GeForce RTX 3090

torch: ver1.9.0

その他必要情報がございましたらご連絡ください。

以下のような質問にはグッドを送りましょう

  • 質問内容が明確
  • 自分も答えを知りたい
  • 質問者以外のユーザにも役立つ

グッドが多くついた質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

気になる質問をクリップする

クリップした質問は、後からいつでもマイページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

下記のような質問は推奨されていません。

  • 間違っている
  • 質問になっていない投稿
  • スパムや攻撃的な表現を用いた投稿

適切な質問に修正を依頼しましょう。

jbpb0

2021/09/09 12:01

> RTX3090+cuda10.2 RTX3090は、下記の「Ampere architecture」なので、現状「CUDA 11.*」のみ対応です https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html 正規の対応をしてない組み合わせで使うと、質問の件が解決して動くようになっても、別の不具合が発生するかもしれません
YuYuMoYuYu

2021/09/09 12:32

あぁ。。。ありがとうございます。。大枚はたいたPCが無に帰すかと。。。 別のエラーが発生して格闘中ですが希望が見えましたありがとうございます

回答1

0

自己解決

Cuda10.2→11.1環境に変更すると解決しました。
ただ,pytorchインストール時にpackageをCondaに設定するとエラーが発生し,Pipにすると解決しました。
原因は不明です。。。

投稿2021/09/11 01:59

YuYuMoYuYu

総合スコア8

良いと思った回答にはグッドを送りましょう。
グッドが多くついた回答ほどページの上位に表示されるので、他の人が素晴らしい回答を見つけやすくなります。

下記のような回答は推奨されていません。

  • 間違っている回答
  • 質問の回答になっていない投稿
  • スパムや攻撃的な表現を用いた投稿

このような回答には修正を依頼しましょう。

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
86.12%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問

同じタグがついた質問を見る

CUDA

CUDAは並列計算プラットフォームであり、Nvidia GPU(Graphics Processing Units)向けのプログラミングモデルです。CUDAは様々なプログラミング言語、ライブラリ、APIを通してNvidiaにインターフェイスを提供します。

PyTorch

PyTorchは、オープンソースのPython向けの機械学習ライブラリ。Facebookの人工知能研究グループが開発を主導しています。強力なGPUサポートを備えたテンソル計算、テープベースの自動微分による柔軟なニューラルネットワークの記述が可能です。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。