ubuntu conda 安装 pytorch-gpu 和 hugging face

0. 已提前安装好显卡驱动

1. 创建环境

conda create -n huggingface python=3.7
conda activate huggingface

2. 给conda换源

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 

源的地址如果使用https会出现CondaHTTPError，因此将https全部转化为http

3. 安装cuda

conda install cudatoolkit=10.0 -c http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

4. 安装cudnn

conda install cudnn=7.6 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

5. 安装pytorch-gpu

 conda install pytorch==1.7.0 torchvision==0.8 cudatoolkit=10.0

注：pytorch版本需要与cuda和cudnn对应

6. 验证是否安装成功

>>> import torch
>>> torch.cuda.is_available()   # 返回True

7. 移除清华源

不移除的话，安装huggingface的时候一直再solving environment

conda config --remove-key channels

7. 安装huggingface

conda install -c huggingface transformers

ref：

Anaconda环境安装GPU版本Pytorch_DORIC.MA的博客-CSDN博客_conda安装pytorch gpu

hugging fance Installation