赞
踩
假定系统已经安装好了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
对于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
安装完成后,可打开python命令行,检验安装是否成功:
import torch
torch.__version__
如果显示版本号,则安装成功。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。