Étapes pour créer PyTorch 1.5 pour CUDA 10.2 sous Windows

Voici les étapes pour créer PyTorch 1.5 pour CUDA 10.2 sous Windows 10. (Au 1er mars 2020) Il comprend également les instructions de construction de torchvision qui sont souvent utilisées avec PyTorch. Les explications détaillées sont omises ici, donc si vous avez besoin d'une explication un peu plus détaillée, veuillez également vous référer à Blog --dev.infohub.cc. J'espère que tu peux.

Créer un environnement

Paramètres tels que les variables d'environnement

--MKL_DEBUG_CPU_TYPE = 5 (Paramètres pour l'utilisation de MKL avec les processeurs AMD) --Le CHEMIN vers d'autres CUDA, etc. est en place

Construire PyTorch, torchvision

Ouvrez l'invite de commande des outils natifs x64 pour VS 2019 dans Visual Studio 2019 et suivez les étapes ci-dessous. (L'explication est publiée sous forme de commentaire)

#Environnement virtuel pour la construction(build_pt)Créer
python -m venv g:\work\build_pt
g:\work\build_pt\Scripts\activate.bat
cd /d g:\work\build_pt

#Mises à jour telles que les packages
python -m pip install --upgrade pip

#Installation des packages requis pour la construction
pip install numpy ninja pyyaml mkl mkl-include cmake cffi wheel

#Obtenir la source
#Contrairement à la procédure sur le site d'origine, le sous-module sera acquis plus tard.
git clone https://github.com/pytorch/pytorch
cd pytorch

# 2020/03/01 13:Passer à l'état au moment du 24 (dernier état essayé cette fois)
git checkout ace2b4f

#Obtenir le sous-module
git submodule sync
git submodule update --init --recursive

#Construire (environ 18 minutes avec Threadripper 3960X)
python setup.py install

#Créer un paquet de roues (mettre la roue dans pip)
# G:\work\build_pt\pytorch\"Torch" en dist-1.5.0a0+ace2b4f-cp38-cp38-win_amd64.whl "est créé
python setup.py bdist_wheel

# ----Construisez la torchvision à partir d'ici----------------------------------

#Dossier parent (g:\work\build_Déplacer vers pt)
cd ..

#Présentation des packages supplémentaires nécessaires pour construire torchvision
pip install six pillow

#Vérifiez si CUDA est utilisé dans la vision de la torche
#Si ce résultat est True, CUDA sera utilisé (vous pouvez également le construire de force, voir la page torchvision pour plus de détails)
python -c "import torch; print(torch.cuda.is_available())"

#Obtenir le code source pour torchvision
git clone https://github.com/pytorch/vision.git
cd vision

git checkout b2e9565

#Construire (quelques minutes)
python setup.py install

#Créer un paquet de roues
python setup.py bdist_wheel

Cela créera un fichier de roue PyTorch et torchvision (* .whl) dans le dossier dist.

Introduction à l'environnement d'utilisation

Il s'agit de la procédure pour installer les packages PyTorch et torchvision wheel intégrés dans des environnements virtuels séparés. Puisque le dossier de construction utilise beaucoup d'espace disque, il est pratique de séparer l'environnement qui utilise réellement PyTorch etc. de l'environnement virtuel pour la construction. Ici, on suppose qu'un environnement virtuel appelé `ml``` est créé sous` c: \ venvs \ ``. Vous pouvez l'exécuter à partir d'une invite de commandes normale (pas de la console Visual Studio)

#Environnement virtuel(ml)Créer
python -m venv c:\venvs\ml
c:\venvs\ml\Scripts\activate.bat
cd /d c:\venvs\ml

#Mises à jour telles que les packages
python -m pip install --upgrade pip

#Présentation de PyTorch et torchvision (y compris les packages dépendants)
#Spécifiez le fichier créé ci-dessus pour le fichier whl de PyTorch et torchvision
pip install numpy mkl six pillow
pip install "torch-1.5.0a0+ace2b4f-cp38-cp38-win_amd64.whl"
pip install "torchvision-0.6.0a0+b2e9565-cp38-cp38-win_amd64.whl"

Ceci termine la création de l'environnement PyTorch 1.5.0a0 pour CUDA 10.2.

Recommended Posts

Étapes pour créer PyTorch 1.5 pour CUDA 10.2 sous Windows
Construire TensorFlow sous Windows
Construisez XGBoost sur Windows
Construisez Linux sur un environnement Windows. Étapes pour installer Laradock et migrer
Comment forcer TensorFlow 2.3.0 à construire pour CUDA11 + cuDNN8
Construire l'extension Python E-Cell 4 sur Windows 7 (64 bits)
F2py sur Miniconda pour Windows
Créer un environnement Python sur Windows
Construire un environnement Python avec Windows
Créer Python3 pour Windows 10 sur ARM avec Visual Studio 2019 (x86) sur Windows 10 sur ARM
Étapes pour déployer EMLauncher sur CentOS 8
Étapes pour installer MySQL 8.0 sur CentOS 8.1
Python 3.6 sous Windows ... et vers Xamarin.
Comment utiliser Dataiku sous Windows
Instructions pour installer matplotlib sur votre Mac
Construisez et installez OpenCV sur Windows
Étapes pour installer VirtualBox sur CentOS
Installez Pytorch sur Blender 2.90 python sous Windows
Comment installer pycrypto sur Windows
Comment déployer django-compresseur sous Windows
Étapes pour installer python3 sur Mac
pykintone sur le sous-système Windows pour Linux
Comment installer Music 21 sur Windows
Que faire si Python ne fonctionne pas sur Git pour Windows
Le moyen le plus simple d'obtenir Chainer v1.5 + CUDA + cuDNN sous Windows
Remarques sur l'utilisation d'OpenCV avec Windows10 Python 3.8.3.
[UE4] Construire DedicatedServer sous Windows et Linux
Environnement de création de module d'extension Python 2.7, 3.4, 3.5 sous Windows
Environnement de programmation pour les débutants réalisé avec Windows
Comment mettre à jour vers Chainer 2.0 (Windows + CUDA 8.0 + CUDNN)
[Kivy] Comment installer Kivy sur Windows [Python]
Comment créer Hello, World avec #Nix
[Note] Procédures d'installation d'Ubuntu sur Windows 10
Procédure de création d'un environnement virtuel Python avec VS Code sous Windows
Étapes pour installer l'environnement Python sur Ubuntu
Comment utiliser l'Assistant Google sur Windows 10
Comment installer le sous-système Windows pour Linux
Création d'un environnement pour "Tello_Video" sous Windows
Comment créer un environnement pour utiliser plusieurs versions de Python sur un Mac
[Forefront of Object Recognition 2020] Un journal de l'installation de Pytorch sur Windows 10 à l'exécution de CornerNet-Lite.
Jusqu'à ce que vous puissiez utiliser RTX 2060 sur Windows 10 (installation du pilote NVIDIA, du kit d'outils NVIDIA CUDA, PyTorch)
Différentes façons d'exécuter des fichiers .py sous Windows
Pour les débutants à créer un environnement Anaconda. (Note)
Mettez MicroPython sur Windows pour exécuter ESP32 sur Python
(Windows) Causes et solutions de contournement pour UnicodeEncodeError dans Python 3
Créez simplement un environnement d'exécution Python 3 sous Windows
[Latest] Comment créer un environnement Java sur Ubuntu
Comment installer rapidement h5py sur Windows 10 [non officiel]
Procédure de génération de TensorFlow 2.2.0-rc0 (CUDA10.2 + cuDNN7.6.5) --Windows10
De l'installation de ROS pour Windows au contrôle de fonctionnement
Comment vivre une vie décente avec Windows 2017
Créer un environnement Kubernetes pour le développement sur Ubuntu
Je veux faire pyenv + pipenv même sous Windows
[Python] Comment installer OpenCV sur Anaconda [Windows]
[Windows] Mémo pour l'utilisation de Keras sur GPU [Tensorflow-GPU]
Convertir les packages binaires pour Windows au format roue
Comment installer / vérifier Graphviz sur anaconda / windows10
Créer un environnement de développement Python basé sur GVim sur Windows 10 (3) GVim8.0 et Python3.6
Comment intégrer mod_wsgi dans Apache sur Windows Python