好得很程序员自学网

<tfoot draggable='sEl'></tfoot>

详解PyTorch 安装指南

创建PyTorch的虚拟环境

在 【最近添加】 里面找到 Anaconda Prompt 并打开,依次输入以下命令:

# 创建名为 PyTorch_envs 的虚拟环境,并指定 Python 版本号为 python 3.8
conda create -n PyTorch_envs python=3.8

# 激活虚拟环境 PyTorch_envs
conda activate PyTorch_envs

# 退出虚拟环境
conda deactivate

创建PyTorch的虚拟环境

在 【最近添加】 里面找到 Anaconda Prompt 并打开,依次输入以下命令:

# 创建名为 PyTorch_envs 的虚拟环境,并指定 Python 版本号为 python 3.8
conda create -n PyTorch_envs python=3.8

# 激活虚拟环境 PyTorch_envs
conda activate PyTorch_envs

# 退出虚拟环境
conda deactivate

验证 PyTorch 是否安装成功

官网 https://pytorch.org/get-started/locally/#windows-verification 也说明了验证方式:



如图所示, PyTorch 安装成功。

安装过程中可能出现的问题

import torch

出现

ImportError: numpy.core.multiarray failed to import

出错原因: numpy 的版本出现问题,导致不能与 PyTorch 的版本进行匹配,具体为 numpy 版本过低。

解决方案:升级 numpy 的版本。激活环境后输入下面的命令:

conda upgrade numpy

End。

以上就是详解PyTorch 安装指南的详细内容!

查看更多关于详解PyTorch 安装指南的详细内容...

  阅读:33次