conda搭建pytorch下的cpu运行环境

聊到深度学习,有各种各样的框架可供使用,比TensorFlow,Keras,PyTorch等,但入门还是有点难度的,经过权衡,这里选择PyTorch作为学习的对象,希望如此打开AI的大门。

由于资源的限制,个人搭建pytorch的环境只能针对cpu类型的,不过作为以后的入门学习研究应该是足够的。

1. 使用conda创建虚拟环境

conda create -n pytorch_env python=3.6

激活虚拟环境

conda activate pytorch_env

2.在虚拟环境pytorch_env中安装pytorch

首先在官网根据配置获取对应的安装命令:

这里根据配置得到了对应的安装命令:

conda install pytorch torchvision cpuonly -c pytorch

等待程序安装完毕:

3.编写测试代码

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

cd到torch_one.py文件对应所在的目录:

cd C:Worksodoo_srcodoo_app

执行测试文件:

python .torch_one.py

最后输出的结果如下:

输出结果为:

hello pytorch1.3.1

程序成功运行并输出结果,到此我们的pytorch环境搭建成功,以后就可以基于此学习AI的知识了,这是一个新的起点。

未经允许不得转载:大自然的搬运工 » conda搭建pytorch下的cpu运行环境

赞 (0)

评论 0

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址