安装pytorch1.10.0+cuda11.4.3+cudnn8.2.1

入门三件套,安装的时候看自己需要哪个版本的,比如我用torchvision0.11.1,目前的最新版,适配pytorch1.10.0,安装的时候如果不安装cuda,就只能在cpu上训练模型。安装cuda的时候要先打开nvdia控制面板,左下角系统信息,组件,看一看自己的驱动支持的cuda版本,我这里是cuda11.4,适配的cudnn版本是8.2.1。

安装pytorch从清华镜像源

pip install torch===1.10.0 torchvision===0.11.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

如果报错,则在-i前面加上-user

cuda可以到nvdia下载对应的版本,这是11.4版,注意网址里是.cn,如果是.com改成.cn这样下就快了,链接在下面。同时我把cuda11.4.3和cudnn8.2.1上传到了百度网盘

链接:https://pan.baidu.com/s/1Y8Ay4dOOdgahBRUQpy_1CA 
提取码:u2f7

cudnn需要登录nvdia开发者会员,但是登录界面被墙了,需要才能下载,需要下载其他版本的cudnn只能或者找了,https://developer.download.nvidia.cn/compute/cuda/11.4.3/local_installers/cuda_11.4.3_472.50_win10.exeicon-default.png?t=LA92https://developer.download.nvidia.cn/compute/cuda/11.4.3/local_installers/cuda_11.4.3_472.50_win10.exe

你可能感兴趣的:(pytorch,深度学习,python)