PyTorchをLinuxにインストールする手順は何ですか。

通常在Linux系统上安装PyTorch需要按照以下步骤进行:

  1. Anacondaをインストールする(未インストールの場合)、対応するシステム用のインストールパッケージはAnaconda公式ウェブサイトからダウンロードできます:https://www.anaconda.com/products/distribution
  2. 新しいconda環境を作成するには、以下のコマンドを使用してください。
conda create -n pytorch_env python=3.7
  1. conda環境をアクティブにするには、次のコマンドを使用してください:
conda activate pytorch_env
  1. PyTorchとTorchVisionをインストールするには、以下のコマンドを使用してください。
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch

CUDAサポートが不要な場合は、以下のコマンドを使用してCPUバージョンをインストールできます。

conda install pytorch torchvision torchaudio cpuonly -c pytorch
  1. インストールを検証するために、以下のPythonコードを使ってPyTorchが正常にインストールされているか確認できます。
import torch
print(torch.__version__)

上記の手順に従うと、LinuxシステムにPyTorchを正常にインストールできるはずです。何か問題がある場合は、PyTorchの公式ドキュメントやコミュニティフォーラムの関連ヘルプを参照することをお勧めします。

bannerAds