不需要安装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版本