首页 > 代码库 > 『PyTorch』第一弹_Linux系统下的安装记录

『PyTorch』第一弹_Linux系统下的安装记录

官网首页(http://pytorch.org/)是有安装教程的,但是点击之后没有反应,原因不明,所以不得不自己寻找一个安装方法。

安装参考如下:

http://blog.csdn.net/amds123/article/details/69396953

由于我的机器使用Anaconda2.7内部嵌套了Anaconda3.6,而我更倾向于使用3.6版本(个人感觉使用3.x是大势所趋,且3.x的确比2.7方便不少),而我的cuda版本是8,所以我根据自己的情况记录一下安装流程:

# 激活环境
source activate python3_6
# 安装torch包
pip install http://download.pytorch.org/whl/cu80/torch-0.1.11.post5-cp36-cp36m-linux_x86_64.whl
# 安装torchvision包
pip install torchvision

实际上Anaconda倾向使用conda,但是报了错,所以转而使用pip命令,安装完成后测试如下:

技术分享

成功!

『PyTorch』第一弹_Linux系统下的安装记录