pytorch基础学习(一) pytorch的安装

1. 下载安装

pytorch是目前应用最宽泛的机器学习库,因其代码框架明晰而广受欢迎。
pytorch的安装也非常简单,进入pytorch官网https://pytorch.org/,可以看到下图:


根据自己的系统版本,安装方式,cuda版本进行选择,直接安装即可。
建议使用conda进行安装,进入anaconda的官网进行下载安装https://www.anaconda.com/

2. 可能出现的问题

因为pytorch的网站是在国外的,国内安装很可能因为网络问题安装不上,此时可以通过添加清华源或从pytorch库网站https://download.pytorch.org/whl/torch_stable.html下载安装包直接安装(注意先装torch,再装torchvision)。

3.安装完成测试

安装完成后,运行以下语句进行测试:
CPU版:

import torch
import torchvision

t = torch.tensor([1, 2, 3], dtype=torch.float32)
print (type(t))  # <class 'torch.Tensor'>
print (t.dtype)  # torch.float32
print (t.device)  # cpu
output:
<class 'torch.Tensor'>
torch.float32
cpu

GPU版:

import torch
import torchvision

t = torch.tensor([1, 2, 3], dtype=torch.float32).cuda()
print (type(t))  # <class 'torch.Tensor'>
print (t.dtype)  # torch.float32
print (t.device)  # cuda:0
output:
<class 'torch.Tensor'>
torch.float32
cuda:0
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容