conda安装cuda_Pytorch安装

最新推荐文章于 2024-08-24 11:45:15 发布

我努力变强

最新推荐文章于 2024-08-24 11:45:15 发布

阅读量3.4k

点赞数

文章标签： conda安装cuda

本文链接：https://blog.csdn.net/weixin_30173361/article/details/112118014

版权

本文介绍了在安装PyTorch前如何核对CUDA版本和驱动版本，解决因版本不匹配导致的问题。提供了通过pip和conda两种安装方式，并强调conda安装能自动处理cudatoolkit，还分享了加速下载的技巧。

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

一、核对版本要求

pytorch版本众多，并且每一个版本所需的cuda版本都不一致。在安装之前需要核对本机的cuda版本和驱动版本。

查看torch对应的cuda版

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

合适的版本：

torch 1.2.0 + cuda 10.0
torch 1.3.0 + cuda 10.1
torch 1.4.0 + cuda 10.1

查看cuda和cudnn版本

# 查看cuda版本
# 方式一
nvcc -V
# 方式二，安装目录查看
cat /usr/local/cuda/version.txt


# 查看cudnn版本
# cudnn里面是一些cpp文件，只能通过安装目录查看
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2