pip 安装 Pytorch

前述

pytorch现在是主流的一个机器学习库，我刚学完tensorflow，得马上转来pytorch学习。

一开始使用pip install pytorch（一般这样，我看名字对了开始下载，也没怀疑有问题，直到报错）

它说少了参数，所以多少得上官网看一眼。

(官网给的中文网站,不是我瞎找的)

官方Pytorch中文手册：【布客】PyTorch 中文翻译

官网：PyTorch

开始安装

安装前记得看下Pytorch对应版本的环境限制,比如python版本,numpy等

官网首页往下拉，直接选择自己要的版本，然后复制底下的pip命令，到你的运行环境就行。

你可以自己修改一下代码，比如指定镜像（这不是本章内容）。

底下有代码和建议

# CPU: 
pip3 install torch torchvision torchaudio
# CUDA 11.7
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
# CUDA 11.8
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

建议：查看自己的电脑显卡是否支持CUDA，如果是私人电脑那就尽量选CPU的版本。

（训练时就白嫖其他资源来计算，因为私人的CPU，GPU资源有限，你也不想闻到自己的显卡的焦香吧。实在不行合伙租一台服务器，直接负责计算，又快又安全，省得内存不够结果爆栈了，节省时间与算力。）

查看是否支持CUDA：

 好像也只有英伟达的显卡支持.像我玩AI画图,软件只支持英伟达显卡驱动,我的缓存还低的一批.

如果是其他的显卡就别想了,老老实实用CPU吧,提供的能力也没英伟达好,还是那句话,去白嫖算力.

步骤:

1. 查看显卡版本

打开设备管理器(自己百度吧) 

2. 查看是否支持CUDA

下方连接,进入界面后往下拉,查看对应版本是否支持CUDA
CUDA GPUs - Compute Capability | NVIDIA Developer

自己找吧

下载CUDA

cuda你还需要到官网去下载，根据自身环境选择exe文件，然后安装即可