深度学习环境
BlackMan_阿伟
不积跬步无以至千里
展开
-
指定使用哪块GPU运行程序
情景:加入实验室有8块卡,前7块卡已经呗占用了,如果我们需要运行程序,肯定会错,错误如下:Traceback (most recent call last): File "train.py", line 182, in <module> fasterRCNN = vgg16(classes, pretrained=True, class_agnostic=args.class_agnostic) File "/home/lthpc/lw/WeaklyObjectDetect原创 2020-07-29 22:06:44 · 2777 阅读 · 0 评论 -
Linux内核更新,找不到显卡驱动解决办法
话说疫情在家,学术还是得搞,奈何服务器重启突然内核升级了,完全找不到显卡驱动了。话不多说,先看出现问题,在说解决办法 。问题:一般要查看CUDA是否正常工作,输入nvidia-smi就可以查看到显卡信息,如图...原创 2020-03-11 17:40:07 · 4642 阅读 · 3 评论 -
搭建深度学习的环境
现在的研究方向是CV,那么必须要搭建深度学习的环境,因此我在折腾了一个中秋假期之后终于把服务器的环境搭建成功了!其中废了不少周折,因此想借此博文来记录一下。环境配置:Ubuntu14+TITAN X+CUDA10.1+CUDNN+Pytorch一、安装显卡驱动因为我的服务器目前只装了一块TITAN X 显卡,因此我必须要装支持我这个显卡的显卡驱动。官网地址:https://www....原创 2019-09-17 20:36:13 · 772 阅读 · 0 评论