PyTorch 环境配置及安装
1. 安装 anaconda
2. 为 PyTorch 创建单独的环境,便于管理。
-
打开 Anaconda Prompt
-
为 PyTorch 创建单独的环境
python 版本与 anaconda 中 python 版本相同即可。
conda create -n pytorch python=3.8
输入 y 开始安装需要的包并创建环境(如果出现了 ERROR 并且你此时正在使用代理,尝试关闭代理)。
创建成功。 -
激活环境并查看环境下的包
conda activate pytorch
pip list
-
如果没有 pytorch 包,需要下载安装。
3. PyTorch 安装
-
pytorch 首页:https://pytorch.org/
选择相关参数,如果电脑没有Nvidia显卡或者只有集显,CUDA 选择 CPU 即可。然后运行下面的指令。 -
在 pytorch 环境下运行指令。
如果安装缓慢,请参考此博客 -
安装检验
命令行输入 python 进入 python 环境,执行import torch
,如果没有报错,说明安装成功。
输入torch.cuda.is_available()
检查 GPU 是否可用, 返回 true 即表示可用。
至此,PyTorch 的安装配置完成。
4. 在新建的 pytorch 环境中安装 Jupyter (可选)
-
进入 pytorch 环境
conda activate pytorch
-
安装需要的包
conda install nb_conda
-
屏幕上显示 done 即安装成功。
-
使用 Jupyter
jupyter notebook