Profil von stdiff

24.12.2020, 00:34

PyTorch mit CUDA funktioniert in PyCharm/IntelliJ nicht?

Ich habe mir gerade PyTorch mit CUDA über Anaconda runtergeladen und wenn ich in das Terminal von Anaconda schreibe:

import torch
if torch.cuda.is_available():
  print('es hat geklappt')

dann gibt er das aus; das heißt das es ja geklappt hat und es mit PyTorch funktioniert.

Wenn ich dann aber in meine IDE gehe (PyCharm und IntelliJ) gehe und den selbe Code schreibe dann gibt er gar nix aus.

Könnte mir jemand bitte erklären wie ich das irgendwie in der IDE zum laufen bekomme?

...zum Beitrag

Hilfreichste Antwort

von stdiff

25.12.2020, 15:46

An deiner Stelle würde ich mal überprüfen, ob ich dieselbe Umgebung verwenden würde. Führe mal die folgenden Zeilen in deiner conda Umgebung und auf PyCharm aus.

import sys
print(sys.executable)

Vermutlich bekommst du unterschiedliche Ergebnisse.

Wenn das der Fall ist, musst du PyTorch (mit CUDA Support) nochmal installieren, aber auf Terminal von PyCharm.

...zur Antwort