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安装成功!