一、创建虚拟环境
切换到conda用户,执行:
conda create -n pytorch python=3.10(pytorch可自行命名)
如后续需要安装 TensorFlow 2.x 版本,Python 推荐使用 3.7 到 3.10 版本。
创建过程:
Proceed输入y,点击回车Enter:
出现done提示:
二、激活环境
conda activate pytorch
出现以下提示:
检测环境是否安装好:
conda info --envs
选择适合自己的Pytorch版本,点击https://pytorch.org/
根据自己的实际情况选择:
复制这句话,输入到控制台执行:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
弹出提示,输入 y:
点击回车Enter:
显示done,完成安装:
三、测试安装成功
输入:python
输入:import torch
执行 exit() 退出之后,执行以下命令查看自己安装的环境:
conda info --envs