pytorch GPU版 安装要点 常见问题

由于本人换了台机器，因此又再安装了一遍pytorch，这一篇是用来记录关键要点的。

常见问题

各版本对应

NVIDIA、cuda与pytorch的版本需要配套，特别是NVIDIA和cuda，需要查阅最新的标准，最好的办法就是直接都更新到最新，肯定没问题。

安装完驱动之后，可以使用nvidia-smi命令确认。

具体版本对应表网站上很多，需要安装的时候最好看一下最新的，或者直接去NVIDIA官网查看。

cudann!!

需要特别注意的是，安装完cuda之后，还需要再将cudann中的相关文件放入cuda，并配置环境变量。相关的文章已经很多，不再赘述。如果不这么做，可能无法卷积。

【2021.10.29 23：41更新！】
paddle下必须配置好cudann，请注意，需要将cudnn中bin下的xxx64.dll文件，放入原cuda的bin文件夹下！！

pytorch中cudatoolkit库与当前cuda版本不对应？

直接按照官网的来。比如说：我现在是cuda11.3，但是pytorch官方给的安装语句是11.1，但是使用后是可行的。小数点后的数字不对应应该影响问题不是很大。

但是一切以官方为准！！

pytorch安装语句？换源？

没必要换成清华镜像啊啥的，直接使用官方的语句并不慢！！（一点几G大概下载了20分钟左右）

目前CUDA11.1及以上的 官方安装语句如下（网址：https://pytorch.org/）：

NOTE: 'conda-forge' channel is required for cudatoolkit 11.1
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

其中官方还特意强调了，使用conda-forge安装通道，是因为要安装cudatoolkit库！因此，如果使用其他源就可能面临着安装失败！！

查看是否安装成功