Linux使用Pytorch（五） Centos7安装Pytorch_pytorch 单机多卡安装 centos

1.配置镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
1
2

2.安装Pytorch

进入官网

执行command

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
1

如果感觉以上仍然很慢，可以尝试用pip工具安装

pip install torch torchvision  -i https://pypi.tuna.tsinghua.edu.cn/simple
1

出现下面这个字样表示安装成功

3.测试

查看Pytorch版本

import torch
print(torch.__version__)
1
2

结果如下表示成功

注：本次使用的是vm虚拟机，所以不能够使用cuda，如果实体机安装cuda后安装pytorch可以使用gpu