入门三件套,安装的时候看自己需要哪个版本的,比如我用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.exehttps://developer.download.nvidia.cn/compute/cuda/11.4.3/local_installers/cuda_11.4.3_472.50_win10.exe