环境 | ubuntu16.04+python3.5+cuda8.0+pip 配置PyTorch

写在前面

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

笔记本

  • 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了哈哈哈~

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容