win10上安装pytorch(cpu版本)
·
如有问题,欢迎指正,如涉侵权,私聊删文
一、安装前所需环境
安装pytorch前,需先安装anaconda和pycharm环境,若已安装可直接跳转至安装pytorch环境部分,请参考win10上安装tensorflow(cpu版本)文章中第一部分安装前所需环境内容。
二、安装pytorch
1、创建并安装pytorch环境
- 启动anaconda prompt。单击启动anaconda命令行,如下图所示:

- 将下载源恢复为默认源。由于清华大学开源软件镜像站暂停了Anaconda镜像服务,因此需使用如下命令将下载源恢复为默认源:
若使用命令后出现如下错误,便是下载源已是默认源,继续后续步骤就行:conda config --remove-key channels

- 将conda更新为最新版本。使用如下命令进行更新:
conda update -n base conda - 创建pytorch-cpu虚拟环境。pytorch-cpu为环境名称,可自行更改,创建命令如下:
conda create -n pytorch-cpu # conda create -n name - 登录pytorch-cpu环境。使用如下命令登录所创建的环境,最前面括号内的内容就是当前所在环境的名称:
conda activate pytorch-cpu # conda activate name - 查看pytorch安装命令,并运行该命令。打开pytorch官网,在该页面查看、复制并运行官网推荐的下载命令,如下图所示:

- 设置pycharm为pytorch运行IDE。请看这个很棒的教程的设置pycharm部分,pytorch和tensorflow是类似的。
2、测试pytorch是否安装成功
- pycharm上创建一个python文件,运行如下代码:
import torch if __name__ == '__main__': print(torch.__version__)
- 输入为torch的版本。代码内容为输出torch版本号,如本文的版本号为1.7.1。

参考文献
更多推荐



所有评论(0)