问题：模型正常在跑，但是经检查没有使用到GPU，发现torch.cuda.is_available()返回False。
解决：发现下载的pytorch版本是cpu版本的，参考文章
添加中科大镜像源：
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
添加镜像之后，直接利用中科大源进行pip下载：
pip3 install torch torchvision torchaudio -i https://pypi.mirrors.ustc.edu.cn/simple/
安装成功后，测试：
python
import torch
torch.cuda.is_available()
若返回True，则说明安装成功。
（在这里要注意，pytorch版本还是要和Cuda版本对应，具体可以自行查询）
（参考作者万川cc，https://www.jianshu.com/p/dc5714189af4）
