在linux中安装pytorch

1. 用conda安装

官网给的教程是说

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

（我的服务器nvidia-smi出来后是11.7版本的）

 尝试之后发现它还是安装的cpu版本的，有人是这样说的，也就是conda安装时自动替换为cpu版本了。

还尝试了网友说的用cudatoolkit，即：

conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch

但也依然没有成功，

import torch
torch.cuda.is_available()  ## 依然是False

所以放弃用conda下载了，可恶！！！

2. 用pip安装

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

以上是官网推荐的下载方式，我的torch环境是用以下命令下载的：

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

下载过程中，torch有大概2G，下载速度很慢很慢，因此，直接在提供下载的网址https://download.pytorch.org/whl/cu116/torch-1.13.1%2Bcu116-cp38-cp38-linux_x86_64.whl将该文件下载下来，并传到 /picb/bigdata/project/myy/torch-1.13.1+cu116-cp38-cp38-linux_x86_64.whl 下,然后用

pip3 install /picb/bigdata/project/miaoyy/torch-1.13.1+cu116-cp38-cp38-linux_x86_64.whl

安装。

安装好torch之后，再继续用

pip3 install torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

安装torchvision和torchaudio。