欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

pytorch安装 + win10 + CPU + pycharm测试

程序员文章站 2022-03-17 08:03:14
...

虚拟环境的创建与**

1.打开anaconda prompt,在命令行中输入如下命令,创建名为pytorch的虚拟环境

#conda create -n pytorch #不指定python版本
conda create -n pytorch python==3.7 #指定环境python版本为3.7

2.在命令行中输入如下命令,**虚拟环境

conda activate pytorch

创建虚拟环境前:
pytorch安装 + win10 + CPU + pycharm测试
创建pytorch虚拟环境成功,并进入后:bash变成虚拟环境名pytorch
pytorch安装 + win10 + CPU + pycharm测试

安装pytorch

方法1:在线安装(下载有点慢,建议方法2
1.进入pytorch官网,获取安装命令
根据自己电脑的情况选择,粘贴下面的命令,在命令行中运行
pytorch安装 + win10 + CPU + pycharm测试
方法2:离线安装
1.下载pytorch安装包
根据方法1获得的命令在命令行中输入后,可以知道自己电脑合适的版本。
备注:
1.这张图是后补的,所以版本显示有问题,实际下载版本为torch-1.3.1+cpu-cp37-cp37m-win_amd64.whl
2.我用官网的命令pip3会报错,所以修改为pip
pytorch安装 + win10 + CPU + pycharm测试
pytorch下载链接下载合适的版本
2.在虚拟环境中打开下载目录,运行如下命令

pip install torch-1.3.1+cpu-cp37-cp37m-win_amd64.whl #pip install + 下载文件名

3.安装成功
pytorch安装 + win10 + CPU + pycharm测试

在pycharm中测试

1.配置解释器,file-settings-projec interpreter
pytorch安装 + win10 + CPU + pycharm测试
此处点apply,其他地方都点ok
pytorch安装 + win10 + CPU + pycharm测试
2.新建python文件输入以下代码

import torch
print("HELLO pytorch {}".format(torch.__version__))

3.运行代码,得到下图,及安装成功
pytorch安装 + win10 + CPU + pycharm测试
4.若安装的为GPU版本,可再输入如下代码,运行后,若返回Ture则表示GPU版本安装成功,返回False则表示没有成功安装GPU版本

print(torch.cuda.is_available())
相关标签: 环境配置