安装pytorch时版本冲突问题

使用conda指令安装超时间等待不超过，故而选择使用pip配合国内镜像下载

激活对应的conda虚拟环境，在管理员权限下终端输入：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple

指令由 官网生成的pip指令+ -i https://pypi.tuna.tsinghua.edu.cn/simple指定清华源

注意：直接使用pip安装可能会出现torch版本和cuda不匹配的问题

可以通过以下代码查看

import torch
import torchaudio
import torchvision

print(torch.__version__) #查看torch版本
print(torchaudio.__version__)
print(torchvision.__version__)
print(torch.cuda.is_available())#查看gpu版本是否可用

'''
Python3.10对应输出为：
    1.13.0+cu116
    0.13.0+cu116
    0.14.0+cu116
    True
'''

官网的下载指令安装的torch与torchvision版本可能不匹配，根据torchvision官网查询到的torch版本找到对应的torchvision版本到download官网地址中下载。

torch	torchvision	Python
main / nightly	main / nightly	>=3.8, <=3.11
2.1	0.16	>=3.8, <=3.11
2.0	0.15	>=3.8, <=3.11
1.13	0.14	>=3.7.2, <=3.10
1.10	0.11	>=3.6, <=3.9

PyTorch	TorchAudio	Python
2.1.0	2.1.0	>=3.8, <=3.11
2.0.1	2.0.2	>=3.8, <=3.11
2.0.0	2.0.1	>=3.8, <=3.11
1.13.1	0.13.1	>=3.7, <=3.10
1.13.0	0.13.0	>=3.7, <=3.10
1.10.0	0.10.0	>=3.6, <=3.9

下载完成会得到一个whl文件，进入目录激活conda环境，pip install 文件名.whl