概述:PyTorch1.0正式版来了,也是学习一波的节奏了。
为什么选择PyTorch???
研究人员及工程师可以轻松利用这一开源深度学习框架的新功能,包括可在 eager execution 和 graph execution 模式之间无缝转换的混合前端、改进的分布式训练、用于高性能研究的纯 C++ 前端,以及与云平台的深度集成
PyTorch 1.0 加速 AI 从原型到生产部署的工作流程。
1.本机环境:CUDA8.0+CUDNN6.0+Anconda3(python3.6),根据各自需求自行安装;
2.从Pytorch(https://pytorch.org/)官网下载对应的whl文件并安装,如下图;
3. 1)建议先下载好进行安装,如下命令:
pip3 install torch-1.0.0-cp35-cp35m-win_amd64.whl
pip3 install torchvision
2)如果直接安装就使用以下命令:
pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp36-cp36m-win_amd64.whl
pip3 install torchvision
3)安装torchvision 可能会报错:
TypeError: parse() got an unexpected keyword argument 'transport_encoding'
解决 方法:
conda install pip
4.测试pytorch gpu版本是否安装成功:
import torch
import torchvision
print(torch.cuda.is_available())
可能会出现的报错:
module compiled against API version 0xb but this version of numpy is 0xa
解决方法:
pip3 install --upgrade numpy
5.如果出现类似的报错,最好更新一下过期的库:
pip3 list
pip3 list --outdated
pip3 install --upgrade 库名
5. 接下来,就开始欢快的Pytorch 学习之路了,小白在路上!!!!!!!!!!!!!!!!
参考资料:[1]PyTorch 1.0 正式版发布了! - 知乎