当前位置:   article > 正文

pytorch Linux环境安装与使用(CPU 与GPU版本)_linux系统最低支持什么版本的pytorch

linux系统最低支持什么版本的pytorch

假定系统已经安装好了Anaconda并配置了python环境,pytorch的安装只需要去pytorch的官方网站查看对应的命令输入即可。

pytorch的官网可点击这里

对于Linux系统CPU版本可以使用以下conda或者pip命令安装(1.11.0版本,其他版本命令可参考官网):

#CPU Only
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cpuonly -c pytorch
pip install torch==1.11.0+cpu torchvision==0.12.0+cpu torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cpu
  • 1
  • 2
  • 3

对于Linux系统的GPU版本可以使用以下conda或者pip命令安装(1.11.0版本,其他版本命令可参考官网):

# CUDA 10.2
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c pytorch

# CUDA 11.3
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

# CUDA 11.3
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

# CUDA 10.2
pip install torch==1.11.0+cu102 torchvision==0.12.0+cu102 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu102
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11

安装完成后,可打开python命令行,检验安装是否成功:

import torch
torch.__version__
  • 1
  • 2

如果显示版本号,则安装成功。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/695677
推荐阅读
相关标签
  

闽ICP备14008679号