Cuda与对应Pytorch安装导流(附cudnn 零积分下载)



总的来说,搭建使用GPU运算的pytorch‘的环境,就是想根据你的显卡,选择对应的CUDA版本,再根据你的CUDA版本选择对应的pytorch的GPU版本


1.CUDA安装参考以下链接

CUDA的安装是包括了:cuda与对应的cudnn的安装,这里的选择是需要根据你显卡的型号来确定的

具体安装参考1

具体安装参考2

为了训练模型,电脑悬搭建了个Cuda和cuDNN的环境参考 以下链接就行,我把cudnn下载下来了如下,大家就不用注册去官网找了,cuda的话太大上传不了到csdn,当然用我这个上传的cudnn的话,cuda要选择10.0版本。

显卡1070: cuda10 与cudnn对应的cuda10
显卡3060: cuda11.3与cudnn对应的cuda11.3 (3060下使用cuda10报错 If you want to use the NVIDIA GeForce RTX 3060 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/,说不支持了…)

注意cudnn环境变量的添加
Cuda与对应Pytorch安装导流(附cudnn 零积分下载)_第1张图片

环境变量参考

cuda的各个历史版本,这里我选的10.0

cuda10.0版本官方链接

cuDNN官网下载链接(也可以直接用下面这个,就不用注册了)

cudnn 对应cuda的10.0的0积分下载

检查是否安装成功

2. pytorch安装参考导流:

pytorch的安装是包括了: pytorch和对应版本的torchvision的安装,而pytorch的GPU版本安装又需要对应你的CUDA的版本

cuda版本对应torch与torchvison版本安装参考1

pytorch和torchvision各版本下载地址传送门

pytorch版本对应关系查看网址

你可能感兴趣的:(python之坑,python)