如何安装pytorch

前面已经介绍了如何用Anaconda Navigator搭建虚拟环境和进行包管理，实际上pytorch也可以在里面进行安装，只是安装的版本是CPU版。如果想要安装GPU版，就需要用pip或conda了。

创建虚拟环境

这一步可以直接在Anaconda Navigator里完成，假设虚拟环境的名称就叫pytorch。

到官网找到安装代码

官网：https://pytorch.org/

启动Anaconda Prompt

安装pytorch

激活pytorch环境，把安装代码拷贝到Anaconda Prompt里面安装即可。

测试

import torch
torch.cuda.is_available()

---

2023年11月10日补充：

在安装pytorch时，选择版本需要注意：

• 如果电脑没有GPU，当然就没有必要安装GPU版；如果有GPU，可以选择GPU版，这样训练起来会更快。
• 如果要发布程序，则要考虑程序的实际使用环境，如果电脑没有GPU，那么同样没有必要安装GPU，因为如果使用GPU版，虽然发布的程序不会报错，但程序包会很大（会比CPU程序包大几倍）。所以，建议构建两个不同的环境，开发时用GPU版，发布时用CPU版。