Jusqu'à ce que vous puissiez utiliser RTX 2060 sur Windows 10 (installation du pilote NVIDIA, du kit d'outils NVIDIA CUDA, PyTorch)

en premier

Je souhaite utiliser le GPU lors de l'apprentissage automatique, mais pour utiliser le GPU, il est nécessaire d'installer un logiciel dédié. Plus précisément, il y avait les deux suivants.

• NVIDIA DRIVER
• Kit d'outils NVIDIA CUDA
• PyTorch

C'est un mémorandum de chaque méthode d'installation.

NVIDIA DRIVER

-Télécharger depuis le site ici. La capacité est d'environ 600 Mo, donc cela prendra du temps.

«Pour mon environnement, j'ai choisi ceux-ci.

• Pour les options d'installation, j'ai choisi Personnalisé. --NVIDIA GeForce Experience était déjà installé, je l'ai donc décoché.
• J'ai fait une installation propre.

Kit d'outils NVIDIA CUDA

-Téléchargez CUDA Toolkit xx (xx est la version) depuis ici.

• Si vous prévoyez d'utiliser PyTorch, veuillez utiliser 10.2 à ce stade.
• J'ai téléchargé la version locale. Il fait plus de 2,5 Go et le téléchargement prend du temps.
• Il s'agit également d'une installation personnalisée.
• Si vous n'avez pas installé Visual Studio, une confirmation vous sera demandée. Si vous prévoyez d'installer Visual Studio, il semble préférable de l'installer à des étapes individuelles.

PyTorch Sélectionnez votre environnement ici. En bas, la commande apparaîtra, alors copiez-la et exécutez-la à l'invite de commande.

Effaçons les fenêtres et mettons ubuntu ...