您好!安装gpu版本的pytorch比较特殊。主要按照以下几步往下走,望对您有所启发和帮助。
1、查看自己显卡型号,下载与自己显卡、系统匹配的cuda和cudnn版本,并安装,安装cudnn时,参考博客,不要把文件拷贝错了(尤其是路径要搞对)
2、下载安装anaconda,配置一个独立的虚拟环境,用于装pytorch,可以先更改源为清华源
3、使用nvcc -v命令,查看cuda和cudnn是否安装成功,安装成功继续操作后面步骤
4、在anaconda创建的虚拟空间下,执行cuda对应pytorch版本安装命令(去pytorch官网查看,命令中同时包含安装torchaudio和torchvision),建议使用pip安装
5、安装完毕,验证
cuda和cudnn都是需要安装的,并且不建议安装在conda环境中,最好单独安装。
pytorch安装详细情看,不仅是torch,还需要安装torchaudio和torchvision,并且版本之间需要匹配