[人工智能]安装pytorch踩坑笔记(已安装cuda和cudnn)

安装pytorch的时候要按照官网的提示执行，不要单独依靠自己的cuda toolkit版本号选择版本。很可能是错误的。

PyTorchhttps://pytorch.org/

官网地址如上图

?

注意我的计算机的cuda的版本是11.5，但是官网的建议是安装11.3

之前尝试11.5安装之后?torch.cuda.is_available()报False说明不支持GPU

安装11.3

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

之后显示

import torch

print(torch.__version__)

x = torch.zeros(5,3,dtype=torch.long)
print(x)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))

?输出

D:\anaconda\envs\torchGPU\python.exe D:/graduation_project/pytorchTry1/main.py
1.10.0
tensor([[0, 0, 0],
        [0, 0, 0],
        [0, 0, 0],
        [0, 0, 0],
        [0, 0, 0]])
True
NVIDIA GeForce RTX 3060 Ti

Process finished with exit code 0

表示支持GPU计算，安装成功

? ? ? ? 其他tips:

? ? ? ? 安装过程中安装conda的cudnn（直接conda install cudnn即可）

? ? ? ? 安装完cudnn之后重启电脑（但是上述安装操作中没有用处，更改cuda tookit版本才解决）