【2021】Pytorch的安装和CUDA问题总结

博主在安装Pytorch的时候，网上的教程鱼龙混杂，有些内容已经过时。

2021年9月，Pytorch（cuda）版本的安装配置，已经完全不需要另外下载CUDA toolkit，因为使用命令行将自动帮你安排好一切。

 Pytorch的极简安装

1. 首先打开Pytorch的官网，点击Get Started。

2.  选择Conda或者Pip安装，并选择合适的CUDA版本（一般来说越新越好，后面会提到）

注意：无论选用Conda还是Pip，它给出的命令行都会默认绑定一个cuda-toolkit，这个cuda-toolkit和你在NVIDIA官网上下载的cuda-toolkit是等效的，也就是说我们不需要重新再去NV官网下载了。

 3. 将命令复制到terminal，等待安装即可。

关于CUDA的问题

之前提到，cuda-toolkit越新越好，这当然是有要求的。这个要求是：你的CUDA驱动版本要高于toolkit的版本。

CUDA驱动的版本是和你显卡驱动的版本息息相关的，可以在CMD窗口输入“nvidia-smi”，查看当前驱动版本。博主在更新显卡驱动后，可以看到CUDA驱动的版本是11.4。

另外，一台主机可以安装多个cuda-toolkit。（下面是博主猜测，还没有实验过）如果使用conda的话，可以在不同的环境中，使用官网给出的命令行，安装不同的cuda版本，这样就省去了win10来回切换cuda版本的烦恼。

参考资料

显卡，显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么？ - marsggbo - 博客园