1.硬件方面首先是检查电脑(服务器)上是否安装了cudnn,cuda。
在安装了的前提下,检查cudnn,以及cuda的版本号:
查看CUDA版本:
cat /usr/local/cuda/version.txt
查看cudnn版本:
cat /usr/local/cuda/include/include/cudnn.h | grep CUDNN_MAJOR -A 2
2.查看torch不同版本发布日期:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
3.查看torchvision不同版本发布日期:
https://pypi.org/project/torchvision/0.2.1/#history
<