GPU
关注
分享
扫一扫
文章平均质量分 79
容花呀_AI
这个作者很懒，什么都没留下…
展开
-
ubuntu16.04 + anaconda + pytorch（或者caffe2）GPU
文章目录1 安装驱动2 安装CUDA和cuDNN3 安装anaconda4 安装pytorch4.1 安装依赖4.2 安装5 给文件夹权限1 安装驱动2 安装CUDA和cuDNN3 安装anaconda4 安装pytorch4.1 安装依赖export CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" # [anacon...原创 2019-01-01 16:35:51 · 355 阅读 · 0 评论 -
caffe2 single gpu for mnist
import numpy as npimport cPickleimport csvfrom matplotlib import pyplot as plt%matplotlib inlinefrom caffe2.python import core,workspace,model_helper,brew,optimizerfrom caffe2.proto import caff...原创 2019-04-27 21:13:50 · 249 阅读 · 0 评论