zl程序教程

您现在的位置是:首页 >  其他

当前栏目

下载安装PyTorch

2023-02-19 12:20:51 时间

1、下载并安装Visual Studio Code选择合适版本安装

2、下载安装conda并配置环境

下载方式一:官网下载

下载方式二:清华镜像安装

3、conda配置环境

打开电脑高级系统配置点开系统环境变量:

找到path然后点击Edit或者直接双击:

之后可以简单通过cmd输入conda命令检查下安装配置是否成功:

  1. 创建pytorch环境

不同的项目需要不同的虚拟环境,可以处理不同版本的项目之间不兼容问题,进入 Anaconda prompt 命令窗口。

PyTorch是虚拟环境名字(可以随意设置),python版本可以按自己需求改,一定要指定具体 python 版本。创建成功后,输入以下命令: conda info –envs

配置清华TUNA镜像源: TUNA 提供了 Anaconda 仓库与第三方源(conda-forge、msys2、pytorch等,查看完整列表)的镜像,各系统都可以通过修改用户目录下的 .condarc 文件。

进入pytorch官网:https://pytorch.org/

官网会自动根据你的电脑,显示的即是你可安装的CUDA版本,并给出安装命令。

检查是否安装成功随便创建一个文件尝试导包,不报错为安装完成

检查是否可用

总结:通过Anaconda创建自己的环境,并且安装torch、torchvision等包,在Anaconda prompt命令窗口下是可以正常导入的,而在PyCharm编译器中则会报错。