pytorch版本升级至1.11.0

近期pytorch版本发布，有较大的改动，所以很有必要将pytorch的版本从原来的1.6.0升级至当前版本1.11.0，笔者所做操作是在Ubuntu20.04系统中，相关操作简单记录如下：

查看pytorch的当前版本：

root@T630-4:/data1# python3
Python 3.8.10 (default, Mar 15 2022, 12:22:08)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
1.6.0
>>> exit()

可以看到，目前所用版本为·1.6.0。所以笔者计划升级pytorch的版本。首先登录pytorch官方网站​​​​​​ Start Locally | PyTorch

 从该网页中，选择相应的选项，例如笔者的相关选项如上图所示，如果无法确定CUDA的版本，可以在ubuntu系统中运行nvidia-smi命令，结果类似如下：

 从上图可以看出CUDA版本为11.6，因此从图1中直接复制最底下的命令行，然后按照上面的cuda版本进行修改，例如将cu113换程cu116

 pip3 uninstall torch torchvision torchaudio
 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

先卸载原来版本的pytorch，然后重新安装pytorch。成功运行之后可以验证一下当前版本信息：

root@T630-4:/data1# python3
Python 3.8.10 (default, Mar 15 2022, 12:22:08)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
1.11.0+cu102