Ubuntu 18.04下安装pytorch

原创于 2018-11-28 17:16:05 发布 · 3.2k 阅读

CC 4.0 BY-SA版权

2 篇文章

订阅专栏

本文详细介绍如何在Linux环境下安装配置CUDA、CUDNN及PyTorch等深度学习框架，包括安装显卡驱动、CUDA、CUDNN，通过Anaconda管理Python环境，以及安装稳定版或最新版PyTorch的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 安装CUDA

包括显卡驱动，CUDA， CUDNN

2. 安装 Anaconda

使用命令行下载安装文件，并安装

curl -O https://repo.anaconda.com/archive/Anaconda3-5.2.0-Linux-x86_64.sh
sh Anaconda3-5.2.0-Linux-x86_64.sh

You may have to open a new terminal or re-source your ~/.bashrc to get access to the conda command.

3. 建立Anaconda环境

建立新环境：conda create --name env_pytorch python=3.6

删除环境： conda remove --name env_pytorch --all

激活环境： source activate env_pytorch

退出环境：source deactivate

4. 安装Pytorch

进入激活环境之后，再安装pytorch及其相关依赖库

安装稳定版本：

   conda install pytorch torchvision cuda92 -c pytorch

安装最新版本：

   conda install pytorch-nightly cuda92 -c pytorch

Ubuntu 18.04下 安装pytorch