接外包,有相关需求的可以联系我:Telegram | Email

PyTorch安装

该文章创建(更新)于07/28/2020,请注意文章的时效性!

缘由

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

步骤

安装CUDA环境

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

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

  • 如为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 torch
print(torch.cuda.is_available())

结果应为: True

  • 简单测试
from __future__ import print_function
import 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]])

参考


要不赞赏一下?

微信
支付宝
PayPal
Bitcoin

版权声明 | Copyright

除非特别说明,本博客所有作品均采用知识共享署名-非商业性使用-禁止演绎 4.0 国际许可协议进行许可。转载请注明转自-
https://www.emperinter.info/2020/07/28/pytorch%e5%ae%89%e8%a3%85/


要不聊聊?

我相信你准备留下的内容是经过思考的!【勾选防爬虫,未勾选无法留言】

*

*



微信公众号

优惠码

阿里云国际版20美元
Vultr10美元
搬瓦工 | Bandwagon应该有折扣吧?
域名 | namesiloemperinter(1美元)