写在前面
梳理一下自己的笔记本和实验室台式机的各自环境~
笔记本
- 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了哈哈哈~
网友评论