Pytorch入门教程-Pytorch安装

Pytorch入门教程-Pytorch安装

说明：

• 介绍如何安装Pytorch安装
• 基于CPU

步骤：

• 使用pip安装
• 为了避免跟其他的python版本冲突，可以直接使用增加新环境

mkdir ~/dl
cd ~/dl/
python3 -m venv venv 
source venv/bin/activate
(venv) $ pip install torch>=1.7.0
(venv) $ pip install torchvision>=0.8.1

• 测试

cd ~/dl/
source venv/bin/activate
(venv) $ python3

• 进入python3交互界面

>>> import torch
>>>