赞
踩
打开pytorch官网,用官网的命令,pip安装第三方库pytorch的时候,
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 i https://pypi.mirrors.ustc.edu.cn/simple/
报错了,
发现是我在pip install的时候,参数i少些了一个“-”。
(1)重新设置临时的Pip源,进行安装,
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn
(2)安装完成之后,测试一下。
- import math
- import torch
- import numpy as np
- import torch.nn as nn
- import torch.optim as optim
- import torch.utils.data as Data
- torch.cuda.is_available()
但是下载下来并不是gpu版本的torch,还是cpu版本的。
(3)重新找了一个命令,我猜测没成功的原因是没加这种“+cu118”后缀。
重新用conda创建了一个虚拟环境,再试一下。
conda create -n dl39 python=3.9
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
这次就可以了,
用命令conda list打印看一下,
OK,成功了。
赞
踩
赞
踩
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。