除非自己有编译需求,无需先安装CUDA和CUDNN,根据官网,只需一条命令
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
在墙内直接下载会非常慢,因此添加pytorch清华源,两行命令搞定
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision torchaudio cudatoolkit=11.3
cudatookit版本按需修改。
有需求,先去官网,csdn的文章少看。大多数水平不太行,多数是互相抄袭,简单的东西,越弄越复杂。
觉得有用,希望点赞评论,顶上去让更多人看到,少走弯路。