记录在2080Ti上安装pytorch0.2

复现vrd-dsr过程中,发现使用pytorch0.2可以使训练的效果随epoch提升,但是根据GitHub上的issue和自己的实验发现使用pytorch0.4或其他大于0.2版本都会使得训练效果随epoch下降,具体原因不明。

下面记录一下如何在个人工作站2080Ti上安装pytorch0.2

1.下载

在官网根据相应环境下载对应wheel安装包

(python2.7)

http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl

(python3.5)

http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp35-cp35m-manylinux1_x86_64.whl

下载torchvision安装包

https://pypi.python.org/pypi/torchvision/0.1.8

2.安装

pip install torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl

pip install torchvision-0.1.8-py2.py3-none-any.whl

3.测试

进入python交互式命令行,输入

import torch

之后还需要安装OpenCV,需要通过conda install opencv命令,不能使用pip install opencv-python==指定版本,这样会改变numpy的版本导致pytorch0.2无法使用。

你可能感兴趣的:(记录在2080Ti上安装pytorch0.2)