cuda+torch安装_cuda torch

作者：空白诗007 | 2024-07-25 13:21:46

踩

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是否匹配

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/空白诗007/article/detail/880403