不需要安装cudatookit和cudnn，只要有显卡就有cuda，安装tenserflow-gpu的时候需要额外安装的是cudatookit和cudnn
用一条命令即可
打开官网：
选择安装版本
运行命令即可
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pytorch默认安装的就是GPU的，如果需要安装CPU的，要加一句cpuonly，参考下面的链接：
Previous PyTorch Versions | PyTorch
如何查看自己的pytorch是gpu还是cpu？
输入：
>>> import torch
>>> torch.cuda.is_available()
若返回为True，则使用的是GPU版本的torch，若为False，则为CPU版本
扫描二维码关注公众号，回复：
17616550 查看本文章
