前提是我的CUDA相关的东西安装好了,因为以前装过tensorflow GPU版本的,如果没有GPU应该也不会影响(我猜的)。现在电脑上没有pytorch软件,参考这里,改了一下torch和torchvision的安装版本,安装步骤:
1、win+R,运行窗口输入cmd打开控制台,输入以下命令
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.6.0 torchvision==0.6.1 -f https://download.pytorch.org/whl/torch_stable.html
等待大约5分钟左右就安装完了,提示安装成功窗口
验证是否安装成功,在该窗口继续输入python 打开python 控制台,输入以下命令检验:
import torch
torch.__version__
显示:1.6.0+cu101
cuda是否可用,接着输入下面命令
torch.cuda.is_available()
返回True