缘由

最近看很多人说Tensoflow，准备转型Pytorch，然后刚好搭建Tensoflow-gpu，搭建好了CUDA环境，就尝试安装了一下！

步骤

安装CUDA环境

具体可以参考https://HdhCmsTestemperinter.info/2020/07/28/tensorflow-gpu-support/

去官网选自己环境配置获取安装代码

https://pytorch.org/get-started/locally/

如为Windows平台+ Pip包管理 + Python语言 + CUDA10.2则会获取到

pip?install?torch===1.5.1?torchvision===0.6.1?-f?https://download.pytorch.org/whl/torch_stable.html

测试

测试CUDA是否可用

import?torchprint(torch.cuda.is_available())

结果应为： True

简单测试

from?__future__?import?print_functionimport?torch
x?=?torch.rand(5,?3)print(x)

结果类似与：

tensor([[0.0730,?0.5183,?0.0066],????????[0.6899,?0.4511,?0.3222],????????[0.6566,?0.9099,?0.4501],????????[0.3694,?0.5197,?0.9324],????????[0.5693,?0.2476,?0.2959]])

查看更多关于PyTorch安装的详细内容...