聊到深度学习,有各种各样的框架可供使用,比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运行环境