机器学习（1）— 开发环境安装

机器学习（1）— 准备开发环境

一、前言

二、Ubuntu开发环境安装

1、NumPy安装

使用如下命令安装：

sudo apt-get install python3-numpy

2、PyTorch 安装

由于我电脑暂时没有英伟达显卡，暂时安装CPU版：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

使用国内的镜像源速度会更快一些：

pip3 install torch torchvision torchaudio -i https://mirrors.aliyun.com/pypi/simple/

使用Python执行以下代码可以验证是否安裝成功

import torch 
print(torch.__version__)          #输出版本号 
print(torch.cuda.is_available()) 

三、Windows开发环境安装