安装win10 安装microsoft visual studio 2015的运行环境(kb4020481),这是安装pytorch的必要条件 安装anaconda3 加入清华源 #由于官方源速度不错,这一步可以不做 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --set show_channel_urls yes 安装pytorch 安装依赖包 conda install numpy mkl cffi conda install pytorch-cpu torchvision-cpu #不能使用CUDA,用CPU做运算 conda install pytorch torchvision #使用CUDA,用GPU做运算 验证PyTorch 安装 在命令行运行python,输入以下代码: from future import print_function import torch x = torch.rand(5, 3) print(x)
输出结果类似于:
tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])