Win11系统/RTX30系列显卡——安装gpu版pytorch

1.电脑显卡:RTX 3060

2.预先安装 cuda11.6(支持win11的最低版本)

参考教程:Windows安装pytorch-gpu_息吹的博客-CSDN博客_window安装pytorch

这个我暂时没有安装

cudnn8.2.2

参考教程:Win10安装CUDA10和cuDNN_码代马的博客-CSDN博客_cuda10.1

3.安装Anaconda3 2021.11

Python 3.9.7

安装conda install pytorch torchvision torchaudio cudatoolkit=11.3(去掉-c后面的代码)

此时安装的时pytorch1.10.1(经测试,pytorch1.2.0+cudatoolkit11.3或pytorch1.2.0+cudatoolkit10.1都失败,可能时因为显卡3060不支持cuda11.1以下的版本&pytorch1.1不支持高版本cuda)

经过n此换源尝试(不同源的顺序很重要,决定能否install成功),最后默认源设置如下

测试pytorch-cuda是否安装成功,输出true则为安装成功

Win11系统/RTX30系列显卡——安装gpu版pytorch_第1张图片

Win11系统/RTX30系列显卡——安装gpu版pytorch_第2张图片

另外,也可以使用pip进行安装,安装3步骤,我是安装失败的,

在pytorch 官网找到对应版本:

Start Locally | PyTorch

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

看下训练模型结果:

Win11系统/RTX30系列显卡——安装gpu版pytorch_第3张图片

你可能感兴趣的:(计算机视觉,pytorch,深度学习,python)