准备工作:
#创建一个名为pytorch的虚拟环境
conda create create -n pytorch
#通过以下语句进行检查:
conda env list
(Pytorch官网现在pip包已经更新为了pip3的下载模式,默认安装最新版本)
下载前应该指定虚拟环境(我指定的是刚创建的pytorch虚拟环境),下载要在此虚拟环境下进行
conda activate pytorch
以下为CPU版本的下载语句,直接复制到命令行里即可:
pip3下载语句
pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
conda下载语句
conda install pytorch torchvision torchaudio cpuonly -c pytorch -i https://pypi.tuna.tsinghua.edu.cn/simple
下载pytorch后用于测试的代码用例:
>python
>>import torch
>>x = torch.rand(5,3)
>>print(x)
因为下载的是CPU版本的pytorch,所以不能使用torch.cuda.is_available()来检验pytorch是否安装成功。