win11用一条命令给anaconda环境安装GPU版本pytorch，并检查是否为GPU版本

leaf_leaves_leaf2024-12-23 22:15

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

输入：

复制代码

>>> import  torch 
>>> torch.cuda.is_available()
 若返回为True，则使用的是GPU版本的torch，若为False，则为CPU版本