如何通过pip install 安装torch的gpu版本和cuda对应

可以通过以下命令来安装torch的GPU版本，并指定CUDA对应的版本：

pip install torch==<版本号>+cu<对应的CUDA版本号> -f https://download.pytorch.org/whl/cu<对应的CUDA版本号>/torch_stable.html

其中，`<版本号>`和`<对应的CUDA版本号>`需要根据实际情况进行替换。例如，如果要安装torch 1.9.0版本，并且CUDA版本为11.1，则可以使用以下命令：

pip install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html

需要注意的是，如果CUDA版本和torch版本不匹配，可能会导致安装失败或者运行时出现问题。因此，在安装之前需要确认好自己的CUDA版本和torch版本的对应关系。