清华源安装tensorflow-gpu及pytorch-gpu

网络上绝大部分的方法要么太复杂，要么没用清华源实在太慢，特此记录了一份。

清华源更换

第一步先换源，linux版本换源

vim ~/.condarc

windows版本换源
参考这个csdn

vim操作简易指南

：
ESC进入更换vim模式系统
I 插入模式
：wq 保存并退出
将原本的内容清除并替换为

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

然后用 conda clean -i 清楚索引。

安装tensorflow-gpu

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.13.1
conda install cudatoolkit==10.0.130-0
conda install cudnn=7.6.5

因为网络原因失败了就多试两次，全网最简版本，有用点个收藏不迷路。

安装pytorch-gpu

方法一、

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

方法二、这个方法有一点邪，前提需要自己安装cuda，不确定以后还能不能用，问题是清华源的conda install一般都不能正确定位gpu版本，需要在安装确实输入y的时候，仔细查看安装包后pytorch是否带有cpu字样，带有cpu证明安装的是cpu版本。

1. 在清华源中找到自己的系统版本，右上角点击时间逆序，在这里以linux64版本为例

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-1.13.1-py3.7_cuda11.7_cudnn8.5.0_0.tar.bz2

2. 下载之后，使用

conda install pytorch-1.13.1-py3.7_cuda11.7_cudnn8.5.0_0.tar.bz2

3. 这种方法过后直接输入

conda install pytorch

4. 再次输入第2步骤

conda install pytorch-1.13.1-py3.7_cuda11.7_cudnn8.5.0_0.tar.bz2

此时需要检查是否为gpu版本，如果带有cuda字样证明安装正确，直接y继续安装。
最后输入下列代码检查gpu是否可用。

import torch
torch.cuda.is_available
torch.zeros(1).cuda()