为了安装pytorch，通常都会直接从官网的命令下载，然而官网的命令通过anaconda的解析，很有可能会安装错版本，这个时候需要我们自己通过pip安装对应的版本。注意，在安装时最好置于外网环境下。

查询cuda版本

通过命令

为11.8。

确认当前环境下python版本

python --version

这里举例为python 3.9

在网址中找到对应pytorch/torchvision/torchaudio版本

访问网址为：

https://download.pytorch.org/whl/{你的CUDA版本}

例如说cuda为11.8，那么就访问：

https://download.pytorch.org/whl/cu118

会进入如下的链接：

其中点击不同的路径可以进入不同的包选择版本，以pytorch为例：
里面有很多内容，可以使用Ctrl+F进入搜索特定版本，例如说torch-1.9.1。

因为我是python 3.9，因此找到cp39，然后linux_x86_64，然后找到文件。

通过命令行安装

也可以通过命令行安装：

pip install torch==1.9.0+cu118 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118

程序会自动寻找最合适的版本，pytorch的版本最好设定一下。