pythonでModuleNotFoundError: No module named 'pycuda._driver' とエラーが出力される

前提

pythonでpycudaを使用しようとした際にエラーが発生しました。

実現したいこと

pycuda.driverをインポートする

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

python
1(仮想環境名) C:\Users\ユーザ名\Test>test.py
2Traceback (most recent call last):
3  File "C:\Users\ユーザ名\Test\test.py", line 1, in <module>
4    import pycuda.autoinit
5  File "c:\Users/ユーザ名/anaconda3/envs/仮想環境名/Lib/site-packages\pycuda\driver.py", line 65, in <module>
6    from pycuda._driver import *  # noqa
7ModuleNotFoundError: No module named 'pycuda._driver'

該当のソースコード

python
1import pycuda.autoinit
2import pycuda.driver as cuda
3import pycuda.gpuarray as gpuarray
4
5print('hoge')
6

試したこと

下記のサイトで該当するwhlファイルをダウンロードしたが、同じエラーが出た。
https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda

下記のコードを実行するとエラーは発生せずに
(2022,1)
と出力される

python
1import pycuda
2
3print(pycuda.VERSION)

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

Python:3.9.12
Anaconda Navigator:4.13.0(こちらの仮想環境のCMD.exe Promptで実行)
GPU:NVIDIA GeForce RTX 3060 Laptop GPU

jbpb0

2022/09/12 02:05

> No module named 'pycuda._driver' https://forums.developer.nvidia.com/t/modulenotfounderror-no-module-named-pycuda-driver/204031/2 を見ると、cudaが使えるように設定した状態でpycudaをインストールする必要があるみたいですが、それは大丈夫でしょうか？上記Webページにはlinuxでの設定方法が書かれてるので、windowsではこちらも見てください https://www.kkaneko.jp/tools/win/pycuda.html

jbpb0

2022/09/12 02:23

> No module named 'pycuda._driver' > Python:3.9.12 c:\Users\ユーザ名\anaconda3\envs\仮想環境名\Lib\site-packages\pycuda に、「_driver.cp39-win_amd64.pyd」みたいな名前のファイルがあるか、確認してみてください (ファイル名は、微妙に違うかも)

jbpb0

2022/09/12 02:38

> 下記のサイトで該当するwhlファイルをダウンロードしたが、同じエラーが出た。 https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda 質問者さんが使ってるcudaのバージョンが分からないので、 https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda から適当に「pycuda-2022.1+cuda116-cp39-cp39-win_amd64.whl」を落として解凍してみたら、中に「_driver.cp39-win_amd64.pyd」が入ってました他のcudaバージョン用のwhlファイルにも、同様に「_driver.cp39-win_amd64.pyd」が入ってるでしょうから、そこから落としたwhlファイルを使ってpipでインストールしたのなら、「_driver.cp39-win_amd64.pyd」が足りないということはないと思いますが、念の為に実際に「_driver.cp39-win_amd64.pyd」が有るのかを確認してみてください

tosa

2022/09/12 02:45

cuda tooolkit v11.7,cuDNN v8.5.0をインストールして、提案していただいた2つ目のリンクの内容をすべて完了しています

tosa

2022/09/12 02:49

_driver.cp39-win_amd64.pydは c:\Users\ユーザ名\anaconda3\envs\仮想環境名\Lib\site-packages\pycudaにありました

jbpb0

2022/09/12 07:02

> 下記のサイトで該当するwhlファイルをダウンロードしたが、同じエラーが出た。 https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda > cuda tooolkit v11.7,cuDNN v8.5.0をインストール https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda には、「cuda 11.7」用のファイル(ファイル名にcuda117が付いてるもの)はありませんダウンロードしてインストールしたwhlファイルのcudaのバージョンと同じバージョンのcudaをインストールしてみてください

tosa

2022/09/12 14:20

cudaのバージョンを合わせて実行したところ、正常に動作しました。自分の相談に解答していただき、本当にありがとうございました。