linux系统安装cuda版本的pytorch

上篇文章中详细记录了安装CUDA的过程，这篇文章继续在上篇文章的基础上记录安装pytorch的过程。

首先我的cuda版本号是11.0，电脑上安装了Anaconda3，python版本为3.6，到pytorch官网上找到对应的版本，首先尝试用conda方式安装：

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

结果报错在现通道上找不到cudatookit=11.0的包由于我的系统上只有一个python版本，用pip安装也一样，故尝试用pip安装方式：

pip install torch1.7.0+cu110 torchvision0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

一番等待之后，显示下载成功。但打开python，输入import torch命令，又报错：
网上找了很久没找到解决办法，然后我又试着直接安装了pytorch：

pip install pytorch

之后再import torch，居然没有报错了，再看一下是否支持cuda：

print(torch.cuda.is_available())

返回True，安装成功。

结果在运行代码时又报torch没有float32错误，应该是pytorch版本问题。