pytorch gpu版本 安装踩坑记录

安装成功了至少五次依旧踩坑，不得已还是记录一下，

网上教程很多，总体来说就是 pip 和conda 安装

conda 安装，坑在于，你在官网复制安装命令，他会给你安装cpu版本，然后你

可以尝试自己解决这个问题，然后你就跳进坑了，

如果你已经按照很多遍依旧不成功，我劝你还是从了PIP 吧，conda安装的坑比较多

安装指令：（cuda11.3版本）

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

pytorch安装地址  历史版本  + 安装指令

https://pytorch.org/get-started/previous-versions/

Start Locally | PyTorch

# 返回当前设备索引
torch.cuda.current_device()


# 返回GPU的数量
torch.cuda.device_count()


# 返回gpu名字，设备索引默认从0开始
torch.cuda.get_device_name(0)


# cuda是否可用
torch.cuda.is_available()

CUDA / CUDA Toolkit和cuDNN的版本对应

未完待续······