本文指导如何在Linux的Anaconda环境中安装PyTorch-GPU，首先确保电脑已装cudnn和cuda，并查看其版本。然后，通过链接查看PyTorch和torchvision的不同版本发布时间，确保torchvision版本早于PyTorch版本以避免不匹配。以PyTorch 0.4.1、CUDA 10和torchvision 0.2.1为例，正确选择对应版本进行安装。
1.硬件方面首先是检查电脑（服务器）上是否安装了cudnn,cuda。
在安装了的前提下，检查cudnn，以及cuda的版本号：
查看CUDA版本：
cat /usr/local/cuda/version.txt
查看cudnn版本：
cat /usr/local/cuda/include/include/cudnn.h | grep CUDNN_MAJOR -A 2
2.查看torch不同版本发布日期：
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
3.查看torchvision不同版本发布日期：
https://pypi.org/project/torchvision/0.2.1/#history
注意：下载的torchvision的版
最低0.47元/天 解锁文章
扫一扫
749
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
