-
仮想環境を作成する
conda create -n pytorch python=3.8
conda env list
conda activate pytorch
-
公式Web サイトにアクセスして、対応するインストール コマンドをダウンロードします:
pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
ただし、ダウンロードは少し遅いです. 最初にMotrix でTorch をダウンロードしました: https://download.pytorch.org/whl/cu101/torch-1.7.0%2Bcu101- cp38-cp38-linux_x86_64 .whl を実行
し、最初に torch をインストールします (torch をダウンロードしたディレクトリに切り替える必要があります):pip install torch-1.7.0+cu101-cp38-cp38-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/
最後に、このコマンドを使用して残りのライブラリをダウンロードします:pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
-
インストールが完了したかどうかをテストします。
import torch import torchvision if __name__ == "__main__": print(torch.__version__) # 查看torch当前版本号 print(torchvision.__version__) # 查看CUDA版本号: print(torch.version.cuda) # 编译当前版本的torch使用的cuda版本号 # 测试Cudnn是否安装成功 from torch.backends import cudnn print(cudnn.is_available()) # 是否GPU可用 print(torch.cuda.is_available()) # 查看当前cuda是否可用于当前版本的Torch,如果输出True,则表示可用
Ubuntu20.04 は Pytorch1.7.0 GPU バージョン torchvision=0.8.1 をインストールします (CUDA と CUDNN がインストールされています)
おすすめ
転載: blog.csdn.net/qq_42887760/article/details/126908666
おすすめ
ランキング