出现这个问题一般是CUDA和CuDNN版本的问题,之前查找解决方案说cuda版本不匹配,但是在pysot项目中,我是按照步骤安装的环境,
conda install pytorch=0.4.1 torchvision cuda90 -c pytorch
所以,需要设置PyTorch的CuDNN基准测试即可,所以在运行程序中添加以下代码:
torch.backends.cudnn.benchmark = True
然后就ok了。
添加的地方如下图所示:
出现这个问题一般是CUDA和CuDNN版本的问题,之前查找解决方案说cuda版本不匹配,但是在pysot项目中,我是按照步骤安装的环境,
conda install pytorch=0.4.1 torchvision cuda90 -c pytorch
所以,需要设置PyTorch的CuDNN基准测试即可,所以在运行程序中添加以下代码:
torch.backends.cudnn.benchmark = True
然后就ok了。
添加的地方如下图所示: