您的位置: 首页 > 文章 > PyTorch+cudnn安装教程

PyTorch+cudnn安装教程

分类: 文章 • 2024-10-19 21:34:46

1. PyTorch安装

先安装anaconda，在安装PyTorch

安装anaconda

去官网找适配系统的版本
bash Anaconda2-5.2.0-Linux-x86_64.sh
然后配置环境变量
vim ~/.bashrc
PyTorch+cudnn安装教程

安装PyTorch

需要匹配python版本，CUDA版本
进入PyTorch官网https://pytorch.org/，
PyTorch+cudnn安装教程

2.cudnn安装
conda install cudnn