pytorch安装.md

来源:互联网 发布:知乎手机版如何写文章 编辑:程序博客网 时间:2024/05/10 21:11

http://pytorch.org/ 官网


安装
If you want to compile with CUDA support, install

  • NVIDIA CUDA 7.5 or above
  • NVIDIA cuDNN v5.x or above

0、先安装完成 cuda(在root环境下安装即可)

参考: TensorFlow安装-Linux篇

1、先使用virtualenv 创建环境(或 使用Anaconda 创建环境)

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ Anaconda下载地址
推荐使用Anaconda(可以自动更改python的版本)

bash Anaconda3-4.4.0-Linux-x86_64.sh conda create -n tensorflow python=3.5.3# 或conda create -n tensorflow python=2.7source activate tensorflow 

再进行安装,即可

2、进入官网 打开下面这个页面,找到对应的安装版本命令

这里写图片描述

pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whlpip install torchvision

如果第一条语句失败,可以按下面语句安装

wget http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whlpip install torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whlpip install torchvision

3、按命令执行安装即可

注:如果Python版本太低,需要手动升级
参考:http://www.cnblogs.com/fx2008/p/6177940.html

原创粉丝点击