PyTorch+cudnn安装教程

1. PyTorch安装

先安装anaconda,在安装PyTorch

安装anaconda

去官网找适配系统的版本
bash Anaconda2-5.2.0-Linux-x86_64.sh
然后配置环境变量
vim ~/.bashrc
在这里插入图片描述

安装PyTorch

需要匹配python版本,CUDA版本
进入PyTorch官网https://pytorch.org/,
PyTorch+cudnn安装教程_第1张图片

2.cudnn安装
conda install cudnn

你可能感兴趣的:(图神经网络,深度学习,机器学习)