快速安装Pytorch以及CUDA匹配问题

pytorch的安装按照官网上来操作是最快速也最稳妥的方式。https://pytorch.org/get-started/locally/

官网非常贴心的给出了各种安装方式，满足各种需求

首先关于python的版本可参见pytorch的github：

需要先安装版本匹配的CUDA以及cuDNN，具体的安装命令如下图操作：

根据自己的需求选择合适的版本，网站就会给出相应的安装命令，目前来看支持CUDA最好的版本就是9.2和10.1

例如上图的选择会返回安装命令：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

在Windows下非常推荐使用Anaconda来配置各种环境，不管是安装还是管理都非常的方便。

安装完成之后可以测试：

扫描二维码关注公众号，回复： 9353866 查看本文章

from __future__ import print_function
import torch

x = torch.rand(5, 3)
print(x)

测试结果：

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

输出与这个结果类似即pytorch安装成功！