See the current Cuda and Cudnn versions for a detailed tutorial on Pytorch0.4.0 environment setup

copyright: this article by [Moneo-ai] original, you big guy, one read, please refer to

statement: as one of the bloggers in AI field, lives up to your time and your

Basic Information

View system information

Cat /etc/issue Ubuntu 16.04.7lts \n \ LCopy the code

View the current Cuda version

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

Copy the code

View the current Cudnn version

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"

Copy the code

start setting up the PyTorch =0.4.0 environment

Step 1 Conda create environment Torch04

Conda create -n torch04 python=3.6.6Copy the code

The following libraries are installed by default

Step 2: Install PyTorch =0.4.0

The installation command is as follows

sourceActivate Torch04 Conda Install PyTorch =0.4.0 -c PytorchCopy the code

The Conda environment for PyTorch =0.4.0 will install CudatoolKit 8.0 required to run Python

Step 3: Install other dependent libraries

PIP install torchVision ==0.1.6 matplotlib opencv-python TQDM scikit-image scipy==0.19.1Copy the code

more effective tutorials

The content of this blog is difficult to generalize, if there are inappropriate places, welcome to the comment area criticism and correction

---

For those who are about to enter the line of computer vision, Murray here recommend the collection of dry goods blog is as follows

• deep learning model training foundation environment construction related tutorial ———— carefully help you sort out
• Computer vision, superfraction reconstruction, target detection, image repair, master/doctor AI communication

Mexic AI

as one of the bloggers with the most dry goods in the field of AI, lives up to his time and qing

If the article is helpful to you, like, comment encourage bloggers every minute to seriously create