pytorch安装及问题

pytorch安装

1.1 在线安装

官网：https://pytorch.org/

在这里选择版本和安装方式。

这里注意一下，CUDA11.7在linux的pip安装中，给出的命令没有指定下载镜像，这里我们要是直接使用官方的命令，下载的就是CPU版本的。
其实pip安装的window和linux命令一样。我们自己加上镜像就可以了。
即：pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117

1.2 离线安装

在线安装一行命令即可。
当然我们也可进入官方的镜像库，使用浏览器下载好whl文件，然后利用pip install xxx.whl离线安装。

进入https://download.pytorch.org/whl/cu117，然后分别进入下面框出的三个文件夹中，选择合适的版本下载。

根据需要选择CPU和CUDA版本。