安装编译的问题
爱护阳光
合作共赢
展开
-
pytorch 指定GPU时,可能出现的问题
一般我们指定GPU训练时,用到的是:os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"今天遇到一个有趣的问题,在我利用以上语句进行指定GPU的时候,上面语句失效,及无论我如何进行指定,都无法实现仅利用‘1,2’块GPU操作,总是利用到原来的0块GPU。**解决方法:于是我将os.environ["CUDA_VISIBLE_DEVICES"] = "1,2"代...原创 2019-12-10 11:23:03 · 1239 阅读 · 4 评论 -
fatal error: caffe/proto/caffe.pb.h: No such file or directory(使用caffe\openpose时编译出错)
https://blog.csdn.net/xiongmingkang/article/details/83421856原创 2019-05-16 14:15:46 · 862 阅读 · 0 评论 -
tensorflow 安装
安装tensorflow时,如果使用直接安装速度相对较慢,采取清华大学的镜像会提高速度。pip3 install tensorflow-gpu==1.8 -i https://pypi.tuna.tsinghua.edu.cn/simple选择版本是1.8,并且是GPU版本...原创 2019-05-14 15:56:45 · 77 阅读 · 0 评论 -
调试神经网络过程中遇到的问题
在这里简单记录一下自己在调试过程中遇到bug:1、可能会遇到不知名的错误,大概就是139电脑内存不足的问题,但是我网络输入比较下,按理说此问题不存在的,后来换到服务器就好了2、另一个就是cuda的问题,需要用的变量都都应该转移到cuda上面,并且与网络是同步的,神经网络中定义的参数也应该单独的转移到GPU上面,因为net.to(device)好像并不会将网络中定义的变量转移到上面。3、最让头...原创 2019-06-29 09:01:02 · 345 阅读 · 0 评论