我的环境:ubutnu14.04。
系统内核:4.13.1-041301-generic
GPU: NVDIA 1080
CUDA:8.0
cudnn 5.1
(CUDA,cudnn之前已经装好)
pytorch所有版本下载地址:点击打开链接
我下载的是:cu80/torch-0.3.1-cp27-cp27mu-linux_x86_64.whl
终端cd到下载的.whl文件的目录
pip2 install cu80/torch-0.3.1-cp27-cp27mu-linux_x86_64.whl
pip2 install torchvison
pytorch 安装结束啦
在pytorch中使用tensorboard,一种方法是用pycrayon这个模块。
pycrayon地址:点击打开链接
但是不要急,注意pycrayon的readme中,要先安装docker。
ubuntu14安装docker,并升级到docker的最新版本(我觉得是必须要新版本的)
一个很好用的安装教程: