1.确定pytorch和cuda的版本
pytorch官网:https://pytorch.org/ 这里选择的是cuda 11.1,torch 1.8.1。(在下载之前要先确定电脑显卡是否支持,向下是兼容的,可以在显卡控制面板的设备信息中找到)
2.下载cuda 11.1。
NVIDIA官网:https://developer.nvidia.com/zh-cn 可以去官网找,推荐直接百度cuda 这是最新版本的cuda,一般不推荐使用。在这里使用的是旧版本的11.1
3.安装和配置cuda
(1)无脑下一步,也可以自主选择 (2)安装完成后,需要注册登录(在这里,网络如果不流畅,可以下载腾讯加速器,对nvidia加速,方便注册和登录) (3)最后win+r,nvcc -V,检查是否安装成功
4.下载cudnn
同样可以去nvidia官网,但还是推荐直接百度cudnn 在这里需要登录注册 注册登录后的界面 选择对应的版本 5.配置cudnn (1)解压下载文件 (2)将cudnn中的bin、include和lib拷贝到cuda对应的目录中去(坑:是拷贝子文件,不是拷贝文件夹) (3)最后win+r,nvcc -V,检查是否安装成功 6.下载和安装pytorch。 推荐方法:将框选的代码粘贴到cmd中,回车即可。(这里推荐添加清华源) 不推荐:如果真的下载速度慢,可以使用迅雷下载,然后再进行安装
|