写在前面

梳理一下自己的笔记本和实验室台式机的各自环境~

笔记本

• ubuntu16.04
• 显卡驱动+cuda8.0+cudnn5.1
• pip管理
• python2.7+python3.5
• 在python3.5下安装了TensorFlow
• 目前想在python3.5下还是用pip管理安装PyTorch

台式机

• ubuntu 14.04
• 显卡驱动+cuda8.0+cudnn5.1
• 原来pip管理下已安装:
  1.TensorFlow
  2.opencv
  3.caffe
• anaconda管理下已安装:
  1.PyTorch 0.3.0
  2.tensorflow 1.3.0

PyTorch安装配置

• 环境说明:
  ubuntu16.04+python3.5+cuda8.0+pip管理
• 安装命令:

sudo pip3 install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp35-cp35m-linux_x86_64.whl 
sudo pip3 install torchvision

• 一般执行安装命令都是需要管理员权限的,记得加sudo哦~~

运行结果

如下图所示,可以使用torch啦~

再也不用困在实验室用torch了哈哈哈~