使用conda安装cuda、torch前要先查看版本

Herene_

已于 2023-06-01 21:56:05 修改

阅读量4.5k

点赞数

分类专栏：安装那些事文章标签： conda pytorch 深度学习

于 2022-12-27 15:19:12 首次发布

本文链接：https://blog.csdn.net/weixin_42165913/article/details/128456216

版权

本文详细介绍了如何在conda环境下安装cuda和pytorch，包括查看系统支持的最高cuda版本，确认python版本，安装pytorch及torchvision，检查pytorch_gpu是否可用，安装cudnn，以及可选的环境变量配置步骤。建议根据cuda版本选择合适的pytorch版本，并注意安装过程中镜像源的选择。

摘要由CSDN通过智能技术生成

1.查看系统算力支持的最高版本cuda

默认系统已经有cuda了，nvcc-V查看现在cuda版本或者nvidia-smi查看cuda驱动支持的最大版本

2.查看conda虚拟环境的python版本

确定torch与python、torch与CUDA、torch与torchvision对的上（torchvision是图像包，不用的可以不安）

torch和pytorch版本对应：https://github.com/pytorch/vision#installation
pytorch各版本和cuda版本对应关系(官网链接)：https://pytorch.org/get-started/previous-versions/

（参考博客：https://www.cnblogs.com/zhoujiayingvana/p/15655016.html）
注意低版本的 pytorch 是否支持更高版本的 cuda。（高版本的pytorch一般能兼容低版本cuda）例如：你需要 1.7.0 的 pytorch，那么 cuda 只能 11.0 及以下。官方推荐的cuda版本为10.2和11.3，这两种 cuda 支持大多数的 pytorch 版本。不过一般人都是根据cuda装pytorch，谁没事指定pytorch版本反装cuda啊，哦是复现baseline啊，那没事了…