Pytorch0.4.0环境搭建之详细教程、查看当前 Cuda和Cudnn版本

• 版权: 本文由【墨理学AI】原创、各位大佬、一文读懂、敬请查阅
• 声明: 作为全网 AI 领域 干货最多的博主之一，❤️ 不负光阴不负卿 ❤️

基础信息

???? 查看系统信息

cat /etc/issue

Ubuntu 16.04.7 LTS \n \l

???? 查看当前 Cuda 版本

nvcc -V

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

???? 查看当前 Cudnn 版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2


#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 3
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

???? 开始搭建 pytorch=0.4.0 环境

???? 第一步 Conda 创建环境 torch04

conda create -n torch04 python=3.6.6

默认就会安装如下库

???? 第二步：安装 pytorch=0.4.0

安装命令如下

source activate torch04

conda install pytorch=0.4.0 -c pytorch

Conda 环境为适配 pytorch=0.4.0 会安装 Python 运行需要的 cudatoolkit 8.0

???? 第三步：安装 其它依赖库

pip install torchvision==0.1.6 matplotlib opencv-python tqdm scikit-image scipy==0.19.1

更多有效教程

此次博文内容难以 以偏概全，如有不恰当的地方，欢迎评论区批评指正

对于即将入行计算机视觉的小伙伴，墨理这里推荐收藏的干货博文目前如下

• ​​❤️ 深度学习模型训练基础环境搭建相关教程————认真帮大家整理了​​
• ​​❤️ 计算机视觉、超分重建、目标检测、图像修复、硕博士 AI 交流​​

墨理学AI

• ???? 作为全网 AI 领域 干货最多的博主之一，❤️ 不负光阴不负卿 ❤️

• ❤️ 如果文章对你有帮助、点赞、评论鼓励博主的每一分认真创作