refer to this link
直接用 pip install torch==0.3.1 安装的是 cuda8.0 的,使用如下命令:
pip install http://download.pytorch.org/whl/cu90/torch-0.3.1-cp35-cp35m-linux_x86_64.whl
安裝前,記得先卸載原先已安裝的torch,再進行安裝。cp35-cp35m,中的35指的是python版本。我用的3.6 python,所以改為cp36-cp36m
refer to this link
直接用 pip install torch==0.3.1 安装的是 cuda8.0 的,使用如下命令:
pip install http://download.pytorch.org/whl/cu90/torch-0.3.1-cp35-cp35m-linux_x86_64.whl
安裝前,記得先卸載原先已安裝的torch,再進行安裝。cp35-cp35m,中的35指的是python版本。我用的3.6 python,所以改為cp36-cp36m