前提
環境構築初心者です。VScodeでcudaを使ってpytorchを動かそうとしたら失敗したため皆様のお力をお借りしたく質問させていただきました。
https://qiita.com/motoyuki1963/items/a334c9488c2f55a867cf
こちらの記事を参考にさせていただきcuda11.7、cuDNN8.7.0で行いました。
pytorchは公式のホームページからcuda、pythonなど自分にあわせてpip3でインストールしました。
pyton3.10、windous10でanacondaは入っております。
実現したいこと
・pytorchをcudaで動かしたい (print(torch.cuda.is_available()) でtrueが出るようにしたい)
発生している問題・エラーメッセージ
False Traceback (most recent call last): File "c:\Users\Maruichi Masahiro\Desktop\VScode\dall e model\.vscode\cu.py", line 4, in <module> torch.cuda.current_device() File "C:\Users\Maruichi Masahiro\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\LocalCache\local-packages\Python310\site-packages\torch\cuda\__init__.py", line 552, in current_device _lazy_init() File "C:\Users\Maruichi Masahiro\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\LocalCache\local-packages\Python310\site-packages\torch\cuda\__init__.py", line 221, in _lazy_init raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled
該当のソースコード
python
1import torch 2 3print(torch.cuda.is_available) 4torch.cuda.current_device()
試したこと
terminalでnvidia-smiを実行してみた結果、cuda11.7となっており、ここから先に進めません。
Tue Dec 20 16:07:17 2022
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 516.94 Driver Version: 516.94 CUDA Version: 11.7 |
|-------------------------------+----------------------+----------------------+
| GPU Name TCC/WDDM | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce ... WDDM | 00000000:B3:00.0 On | N/A |
| 39% 36C P8 17W / 215W | 522MiB / 8192MiB | 9% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 3816 C+G ...y\ShellExperienceHost.exe N/A |
| 0 N/A N/A 6440 C+G C:\Windows\explorer.exe N/A |
| 0 N/A N/A 8188 C+G ...5n1h2txyewy\SearchApp.exe N/A |
| 0 N/A N/A 8232 C+G ...icrosoft VS Code\Code.exe N/A |
| 0 N/A N/A 9040 C+G ...cw5n1h2txyewy\LockApp.exe N/A |
| 0 N/A N/A 10000 C+G ...e\PhoneExperienceHost.exe N/A |
| 0 N/A N/A 11624 C+G ...perience\NVIDIA Share.exe N/A |
| 0 N/A N/A 11984 C+G ...2txyewy\TextInputHost.exe N/A |
| 0 N/A N/A 12972 C+G ...\app-1.0.9008\Discord.exe N/A |
| 0 N/A N/A 13244 C+G ...ge\Application\msedge.exe N/A |
| 0 N/A N/A 13528 C+G ...8wekyb3d8bbwe\Cortana.exe N/A |
| 0 N/A N/A 15144 C+G ...lPanel\SystemSettings.exe N/A |
| 0 N/A N/A 16084 C+G ...bbwe\Microsoft.Photos.exe N/A |
| 0 N/A N/A 16212 C+G ...5n1h2txyewy\SearchApp.exe N/A |
+-----------------------------------------------------------------------------+
補足情報(FW/ツールのバージョンなど)
pip list はこのようになっています。
Package Version
absl-py 1.3.0
aiohttp 3.8.3
aiosignal 1.3.1
antlr4-python3-runtime 4.9.3
appdirs 1.4.4
asttokens 2.2.1
astunparse 1.6.3
async-timeout 4.0.2
attrs 22.1.0
backcall 0.2.0
cachetools 5.2.0
certifi 2022.12.7
charset-normalizer 2.1.1
colorama 0.4.6
coloredlogs 15.0.1
decorator 5.1.1
einops 0.6.0
executing 1.2.0
filelock 3.8.2
flatbuffers 22.12.6
frozenlist 1.3.3
fsspec 2022.11.0
ftfy 6.1.1
gast 0.4.0
google-auth 2.15.0
google-auth-oauthlib 0.4.6
google-pasta 0.2.0
grpcio 1.51.1
h5py 3.7.0
huggingface-hub 0.11.1
humanfriendly 10.0
idna 3.4
imhist 0.0.4
imWatermark 0.0.2
invisible-watermark 0.1.5
ipython 8.7.0
jedi 0.18.2
keras 2.11.0
libclang 14.0.6
lightning-utilities 0.4.2
Mako 1.2.4
Markdown 3.4.1
MarkupSafe 2.1.1
matplotlib-inline 0.1.6
mpmath 1.2.1
multidict 6.0.3
numpy 1.24.0
oauthlib 3.2.2
omegaconf 2.3.0
onnx 1.13.0
onnxruntime 1.13.1
opencv-python 4.6.0.66
opt-einsum 3.3.0
packaging 22.0
parso 0.8.3
pickleshare 0.7.5
Pillow 9.3.0
pip 22.3.1
platformdirs 2.6.0
prompt-toolkit 3.0.36
protobuf 3.19.6
pure-eval 0.2.2
pyasn1 0.4.8
pyasn1-modules 0.2.8
pycuda 2022.2
Pygments 2.13.0
pyreadline3 3.4.1
pytools 2022.1.13
pytorch-lightning 1.8.5.post0
PyWavelets 1.4.1
PyYAML 6.0
regex 2022.10.31
requests 2.28.1
requests-oauthlib 1.3.1
rsa 4.9
six 1.16.0
stack-data 0.6.2
sympy 1.11.1
tensorboard 2.11.0
tensorboard-data-server 0.6.1
tensorboard-plugin-wit 1.8.1
tensorboardX 2.5.1
tensorflow 2.11.0
tensorflow-estimator 2.11.0
tensorflow-intel 2.11.0
tensorflow-io-gcs-filesystem 0.29.0
termcolor 2.1.1
torchaudio 0.13.1+cu117
torchmetrics 0.11.0
tqdm 4.64.1
traitlets 5.7.1
transformers 4.25.1
typing_extensions 4.4.0
urllib3 1.26.13
watermark 2.3.1
wcwidth 0.2.5
Werkzeug 2.2.2
wheel 0.38.4
回答1件
あなたの回答
tips
プレビュー