[PYTHON] Bis Sie RTX 2060 unter Windows 10 verwenden können (NVIDIA DRIVER, NVIDIA CUDA Tool Kit, PyTorch installieren)
zunaechst
Ich möchte beim maschinellen Lernen die GPU verwenden, aber um die GPU verwenden zu können, muss dedizierte Software installiert werden. Insbesondere gab es die folgenden zwei.
- NVIDIA DRIVER
- NVIDIA CUDA Tool Kit
- PyTorch
Es ist ein Memorandum jeder Installationsmethode.
NVIDIA DRIVER
-Download von der Website hier.
Die Kapazität beträgt ca. 600 MB, daher wird es einige Zeit dauern.
――Für meine Umgebung habe ich diese ausgewählt.
- Für die Installationsoptionen habe ich Benutzerdefiniert gewählt.
--NVIDIA GeForce Experience wurde bereits installiert, daher habe ich es deaktiviert.
- Ich habe eine saubere Installation gemacht.
NVIDIA CUDA Tool Kit
- Laden Sie das CUDA Toolkit xx (xx ist die Version) von [hier] herunter (https://developer.nvidia.com/cuda-toolkit-archive).
- Wenn Sie PyTorch verwenden möchten, verwenden Sie zu diesem Zeitpunkt bitte 10.2.
- Ich habe die lokale Version heruntergeladen. Es ist über 2,5 GB groß und das Herunterladen dauert einige Zeit.
- Dies ist auch eine benutzerdefinierte Installation.
- Wenn Sie Visual Studio nicht installiert haben, werden Sie zur Bestätigung aufgefordert. Wenn Sie Visual Studio installieren möchten, ist es besser, es in einzelnen Phasen zu installieren.
PyTorch
Wählen Sie Ihre Umgebung unter [hier] aus (https://pytorch.org/get-started/locally/).
Unten wird der Befehl angezeigt. Kopieren Sie ihn also und führen Sie ihn an der Eingabeaufforderung aus.
Lassen Sie uns Fenster löschen und Ubuntu setzen ...