Étapes de configuration de Pytorch sur CentOS

En déploiement de Pytorch sur le système CentOS, cet article fournit des étapes détaillées.

1. Installation d'Anaconda

1. Allez sur le site officiel d'Anaconda pour télécharger le package d'installation d'Anaconda pour CentOS.
2. Exécutez le programme d'installation et suivez les invites pour terminer l'installation. Il est recommandé de sélectionner la méthode d'installation "Just Me" pour faciliter la configuration des variables d'environnement ultérieure.

2. Création de l'environnement virtuel

1. Ouvrez le terminal, créez un environnement virtuel nommé pytorch_env à l'aide de la commande suivante et spécifiez la version python (comme Python 3.10):

 conda crée -n pytorch_env python = 3.10

2. Activez l'environnement virtuel:

 conda activer pytorch_env

3. Configuration de la source miroir (facultative)

Afin d'accélérer les téléchargements, il est recommandé de configurer une source de miroir domestique. Modifiez le fichier .condarc (généralement situé dans le répertoire personnel de l'utilisateur) et ajoutez ce qui suit:

 <code>channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forges/ show_channel_urls: true auto_activate_base: false</code>

4. Installation de Pytorch

1. Vérification de la version CUDA: utilisez la commande suivante pour afficher la version CUDA:

 nvidia-SMI

Si CUDA n'est pas installé, veuillez l'installer en fonction de vos besoins.

2. Téléchargement et installation de Pytorch:

   • Visitez le site officiel de Pytorch ou le site de miroir de logiciel open source de l'Université Tsinghua pour obtenir la version Pytorch requise.
   • Sélectionnez la version Pytorch qui correspond à la version CUDA. Par exemple, CUDA 12.1 correspond à la version CUDA 12.1 de Pytorch.

   Installez à l'aide de pip (prendre CUDA 12.1 comme exemple):

 PIP Installer Torch TorchVision Torchaudio - index-url https://download.pytorch.org/whl/cu121

Ou installer en utilisant conda :

 Conda Installer Pytorch Torchvision Torchaudio Cudatoolkit = 12,1 -c Pytorch

3. Vérification de l'installation: Après l'activation de l'environnement virtuel, entrez l'environnement interactif Python et exécutez la commande suivante pour vérifier l'installation:

 Importer une torche
Imprimer (torche .__ Version__)
print (torch.cuda.is_available ())

torch.cuda.is_available() renvoie True pour indiquer que Pytorch a été installé avec succès et peut utiliser le GPU.

5. Investigation de problème

• Numpy Conflits with Torch Version: Si la version Numpy et Torch est incompatible, essayez de rétrograder Numpy:

 PIP installe "Numpy <num de version>"</num>

• Commande nvcc INTÉRIEURE: La commande nvcc non trouvée peut être due à l'erreur d'installation de CUDA ou aux variables d'environnement non configurées correctement. Veuillez vous assurer que le chemin d'installation de CUDA a été ajouté aux variables d'environnement du système.

Reportez-vous à la documentation officielle de Pytorch ou au forum communautaire pour résoudre d'autres problèmes rencontrés pendant le processus d'installation. J'espère que les étapes ci-dessus peuvent vous aider à configurer en douceur l'environnement Pytorch sur le système CentOS.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!