tensorflow可以使用GPU，而torch无法使用GPU

具有conda测试环境无意使用pip更新了tensorflow后，导致总报提示"could not load dynamic library 'libnvinfer.so.7'"，经过查询后建议pip uninstall tensorflow后使用conda安装。

conda install tensorflow后解决了库动态加载问题，但是torch出现无法访问GPU的问题。

卸载torch而后重装，可以恢复GPU。

1）查看cuda的版本：cuda list cuda

2）从pytorch官网查找安装命令：

# CUDA 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

3）验证torch能否访问GPU:

import torch

print (torch.cuda.is_available())

参考链接：

1、pytorch 安装命令：Previous PyTorch Versions | PyTorch

2、Could not load dynamic library 'libnvinfer.so.7' 解决方法，Could not load dynamic library 'libnvinfer.so.7' 解决方法-CSDN博客