今天重装系统,然后出了一些问题,也解决了很多问题,这里记录一下!
1.GPU的驱动要记得升升级,要不cuda支持的最高版本低的话,下载了低版本的cuda会出现很多软件或者啥的不合适问题,比如darknet vs2019 windows10这套组合的话,你用cuda10.1进行编译会出现C1189&MSB372这种错误,都是版本的问题
2.一个电脑可以两个cuda版本共存,只需要设置让哪一个生效即可,环境变量的设置,可以看学长博客,有详细配置
3.还有就是nvdia developer这个网站下载cudnn需要登陆,建议把自己的nvdia账号和微信号链接一下,就会方便太多!
4.下载torch,torchvision需要注意cuda匹配关系,最稳妥的方法是去https://download.pytorch.org/whl/torch_stable.html下载,不要pip install torch,这样下载下来的torch你会发现torch.cuda.is_available()是False,在上面这个网站你需要下载两个文件,torch和torchvision。选择相应的python的版本(eg:cp36代表python3.6),torch版本(cu101代表cuda 10.1)。
这个网站下载速度很慢,我建议的方式是在windows上用neatdownloader下载,然后用U盘拷到自己的电脑上。(下载很慢的话暂停继续下,直到速度很快拉满就可以了)
安装torch0.4的时候在https://download.pytorch.org/whl/torch_stable.html
找相应的gpu版本,我发现最高支持的cuda版本是cuda9.2,所以安装一个cpu版本的torch0.4吧!
而且对于windows系统来说,torch0.4最少支持python3.5而linux系统可以python2.7