1.找好pytorch与cuda的对应关系.这个链接可以查对应cuda的安装包
2.找好pytorch与torchvision的对应关系
2.查好自己的GPU与驱动与CUDA和CUDNN的对应关系
3.找好对应CUDA的pytorch安装方法
4.当conda,pip安装较慢时,将安装包下载下来本地安装,对于linux
1.找好pytorch与cuda的对应关系.这个链接可以查对应cuda的安装包
2.找好pytorch与torchvision的对应关系
2.查好自己的GPU与驱动与CUDA和CUDNN的对应关系
3.找好对应CUDA的pytorch安装方法
4.当conda,pip安装较慢时,将安装包下载下来本地安装,对于linux