第一次接触Pytorch 时是17年末,当时Pytorch官方还没有支持windows平台。一般是使用windows+Anaconda 的解决方案
最近需要再装一台电脑,偷懒,想直接利用17年的方法安装Pytorch,结果程序运行报错:
RuntimeError: inconsistent tensor size at d:\downloads\pytorch-master-1\torch\lib\th\generic/THTensorMath.c:832
代码定位显示,错误出现再Pytorch读取数据时
for i, (input, target) in enumerate(loaders[split]):
因为这套程序是之前运行过,跑出结果的。因此猜测是Pytorch安装在某个方面出现了问题。
于是卸载Pytorch重新安装
conda uninstall pytorch
conda uninstall libtorch
使用Pytorch官方给的windows安装方案:https://pytorch.org/get-started/locally/
- 首先安装Anaconda3
- 安装Pytorch
Pytorch官方根据你所选择版本以及系统提供安装命令
在Anaconda Prompt中输入该命令即可完成安装。
如果电脑是AMD的显卡,则安装CPU版本的
conda install pytorch-cpu torchvision-cpu -c pytorch
如果想选择其他安装版本的
# Installing with CUDA 9
conda install pytorch=0.4.1 cuda90 -c pytorch
# 或者
conda install pytorch=0.4.1 cuda92 -c pytorch
# Installing with CUDA 8
conda install pytorch=0.4.1 cuda80 -c pytorch
# Installing with CUDA 7.5
conda install pytorch=0.4.1 cuda75 -c pytorch
# Installing without CUDA
conda install pytorch=0.4.1 -c pytorch
至此,Pytorch安装完毕,
测试是否安装成功
打开Anaconda Prompt,进入python
输入
import torch
不报错,则说明安装正常
# 打印 Pytorch 版本
print(torch.__version__)