Pytorch 0.2.1をバージョンダウンする方法

困っていること

下記の問題のgithubでget_dataset.py、data_util.py、train.pyを実行した結果、

RuntimeError: dimension specified as 0 but tensor has no dimensions

と出力されます。

get_dataset.py
では、
re.findall("<p>.*</p>", html)の部分を

re.findall("<p>.+?</p>", html, re.MULTILINE | re.DOTALL)

と直す必要があるようです。

エラーが出ているgithubのリンク

githubを掲載しているサイト

過去のteratailで問題のgithubを質問したリンク

pytorchのバージョンダウンをすすめる例

このような問題を解決するため、pytorchのバージョンダウンを行いたいと考えています。
しかし、pytorchは、torchとtorchvisionの両方があるため、どのようにインストールすればよいのか分かりません。
できれば、pipまたは、pip3でインストールを行いたいです。

何卒、よろしくお願いいたします。

環境

Ubuntu 18.04.1 LTS (x86_64)
Python 3.6.6
torch 0.4.1
torchvision 0.2.1
mecab-python3 0.7