Ubuntu的pip安装pytorch

1. 显卡驱动要装好，确保nvidia-smi可以运行

2. 推荐在conda环境中用pip进行安装，conda环境配置可以看下述博客

Ubuntu安装miniconda3_tiantangzhixia的博客-CSDN博客Ubuntu安装miniconda3_ubuntu安装miniconda3https://blog.csdn.net/baidu_35280192/article/details/127530088?csdn_share_tail=%7B%22type%22%3A%22blog%22%2C%22rType%22%3A%22article%22%2C%22rId%22%3A%22127530088%22%2C%22source%22%3A%22baidu_35280192%22%7D

3. 进入到conda虚拟环境中，安装torch环境

截至到目前的最新版是torch2.0.1，cuda11.8。虽然我的显卡驱动自带的cuda是12.0但也可以安装，推荐采用python3.10，因为当前的torch.compile只支持到python3.10。

conda activate torch2.0
conda install cudatoolkit cudnn
pip install torch==2.0.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
pip install torchmetrics tensorboard tensorboardX

4. 检查是否可以调用gpu

python
import torch
torch.cuda.is_available()

如果是True，那就ok啦

如果torch.cuda.is_available()的结果为False，那就需要覆盖一下torch，虽然大概率不会有这种情况，但不排除中彩的......

pip install --force-reinstall torch==2.0.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118