安装 pytorch

安装 pytorch

1、单击启动 Anaconda Prompt 创建虚拟 “房间”

通过 conda 创建一个名为：pytorch的虚拟 “房间”，可以方便未来对不同版本的 PyTorch 进行管理，具体方法如下：

conda create -n pytorch python=3.6

---

2、查看创建的 “房间”

conda info --envs

---

3、安装 PyTorch

3.1、激活 pytorch 房间

conda activate pytorch

当前面从（base）变为（pytorch）时表示此时已经切换到 pytorch 房间，随后正式进入安装 pytorch 环节。

3.2、安装 pytorch

https://pytorch.org/get-started/locally/ 获取安装命令。

安装后的文件位于 D:\anaconda\pkgs\ 文件夹内。

3.3、验证

选择正确的解释器后导入模块：

import pytorch

4、配置 Jupyter Notebook 使用Pytorch

先打开虚拟环境：

conda activate pytorch

安装必要的插件：

conda install nb_conda

conda install ipykernel

---

补充：5、在 Linux 上安装 PyTorch

与之前步骤完全一样，不过需要注意权限的问题：

sudo chmod 777 -R ~/anaconda3/
sudo chmod 777 -R ~/.conda/