cuda+torch安装

1. 查看torch版本

python -c "import torch; print(torch.__version__)"

2. 查看cuda版本

python -c "import torch; print(torch.version.cuda)"

3、查看LInux系统配置

nvidia-smi

CUDA _VISBLE=0 PYTHON

4、安装cudatoolkit    cudnn

要写 CUDA 的程序，只需要安装 CUDA Toolkit 即可

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

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

5、conda存在自动升级torch的情况，怎么降级版本呢？离线安装最稳定。

1）查看目前cuda和driver的版本是否匹配

 

2）两者匹配，查看 torch是否匹配

被conda升级了，那就需要降级。

3）到官网找到资料包下载https://download.pytorch.org/whl/torch_stable.html 

 4）上传服务器后离线安装。

pip install /home/.../torch-1.7.1+cu101-cp36-cp36m-linux_x86_64.whl
pip install /home/.../torchvision-0.8.2+cu101-cp36-cp36m-linux_x86_64.whl

比在线方便多了