遇到的问题
接触pytorch 有一段时间了,一直想简单明了的编程环境 不想安装太多臃肿的东西。 以前一直是以pytorch cpu编程 简单的实现 发现这条路行不通。在实现 faster-rcnn 后发现ROI layer 必须是在gpu上跑 没有cpu版本 。所以还是 老老实实安装 pytorch gpu版本。
所以得安装 几个东西: 1, 显卡 驱动 2,cuda 3, cudnn 4,pytorch
安装过程
安装显卡驱动
这个用驱动精灵 或者 鲁大师 都行
安装cuda
https://developer.nvidia.com/cuda-downloads 英伟达官网自动检测你的显卡 版本。
历史版本:
https://developer.nvidia.com/cuda-toolkit-archive
安装 cudnn
从这下载 官网:https://developer.nvidia.com/cudnn 这个网站 需要注册 。 一步一步验证。
这个网站有时候下不懂 ,用火狐浏览器下载 比较好。
相比标准的cuda,它在一些常用的神经网络操作上进行了性能的优化,比如卷积,pooling,归一化,以及激活层等等。
(1)解压:会生成cuda/include、cuda/lib、cuda/bin三个目录;
(2)分别将cuda/include、cuda/lib、cuda/bin三个目录中的内容拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5对应的include、lib、bin目录下即可。
(3)将bin所在的目录添加到环境变量 PATH 中,“此电脑”→“高级系统设置”→“环境变量”→“系统变量”→“path”→“编辑”→“新建”加入该路径即可。
安装pytorch
官网: https://pytorch.org
我喜欢直接用pip 安装 拷贝命令:
pip3 install torch=1.3.1 torchvision=0.4.2 -f https://download.pytorch.org/whl/torch_stable.html
输入cmd 安装完成。
选择conda
1,安装conda
2,运行
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
3,cuda 不容易安装上
4,找到提示网址
<https://repo.anaconda.com/pkgs/main/win-64/cudatoolkit-10.1.243-h74a9793_0.conda
离线下载
5,用这个命令
conda install --use-local cudatoolkit-10.1.243-h74a9793_0.conda 所在的目录