锐客网

  1. 首页 > it技术 > >

Win10+Anaconda|Win10+Anaconda 环境下安装 PyTorch

1.安装Anaconda

参见https://www.jianshu.com/p/e61a33e62bf3
2.创建环境
conda create --name pytorch python=3.6

这里的 pytorch 是虚拟环境的名称，可随意取
激活环境pytorch
activate pytorch

3.安装pytorch 设置清华的下载源
参见 https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
否则可能出现下载慢的情况，编辑C:\Users[你的用户名]\.condarc文件，替换为以下内容
channels: - defaults show_channel_urls: true default_channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r custom_channels: conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

安装pytroch
参见 https://pytorch.org/get-started/locally/
Win10+Anaconda|Win10+Anaconda 环境下安装 PyTorch
文章图片
image.png
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

4.安装cudatoolkit
参见 https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exenetwork
我们选择好Windows版本，然后选择网络安装或者本地安装都可以，然后点击下载安装
Win10+Anaconda|Win10+Anaconda 环境下安装 PyTorch
文章图片
image.png 5.验证 为了确保pytorch安装正确，我们可以运行示例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]])

此外，要检查Pytorch是否启用了GPU驱动程序和CUDA并可访问，请运行以下命令以返回CUDA驱动程序是否已启用
import torch torch.cuda.is_available()

【Win10+Anaconda|Win10+Anaconda 环境下安装 PyTorch】输出是True代表已启用

    推荐阅读


    上一篇：2020我在残酷的日子里，明白了最简单的事

    下一篇：真假“垃圾”大搜索！——《扔或不扔》420