可以使用Anaconda来创建PyTorch的虚拟环境。下面是在命令行中运行的示例代码:
-
首先打开Anaconda Prompt(或者任何支持命令行操作的终端);
-
输入以下命令来创建名为"myenv"的新环境并安装Python版本3.8:
conda create -n myenv python=3.8
这将会自动安装最新版本的PyTorch及其依赖项。
-
等待安装完成后,通过以下命令激活该环境:
conda activate myenv
-
现在已经进入了名为"myenv"的PyTorch环境,可以验证是否正确安装了PyTorch:
import torch print(torch.__version__)
如果能够成功导入
torch
模块并显示当前安装的PyTorch版本号,则表明环境配置成功。 -
若要退出该环境,只需再次运行以下命令:
conda deactivate