当前位置：首页 > CN2资讯 > 正文内容

Windows10下PyTorch环境安装windows10 安装

14小时前CN2资讯

一、CUDA安装

桌面点击右键，查看NVIDIA控制面板

查看系统信息

查看版本

登录nvidia官网查看选择cuda的安装版本

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

我的显卡只能安装CUDA 10.1在高的版本就无法安装了

下载CUDA 10.1，需要注册一个账号，如果无法访问，请使用绕墙软件

官网：https://developer.nvidia.com/cuda-10.1-download-archive-update2

下载安装完毕，在命令行下输入 nvcc -V, 出现下列信息说明Cuda安装成功

二、CuDNN安装

下载CuDNN

官网 https://developer.nvidia.com/rdp/cudnn-download

注意版本不要选高了

复制到CUDA的安装目录，我的安装目录如下

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

复制后

测试，使用管理员运行cmd，进入NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite目录，运行bandwidthTest.exe，显示PASS则说明安装成功

三、Anaconda安装

下载链接：https://www.anaconda.com/

选择版本

一路下一步将其安装

查看python版本

四、Anaconda Prompt配置

单击启动Anaconda Prompt 创建虚拟“房间”

通过conda创建一个名为：pytorch的虚拟“房间”，可以方便未来对不同版本的PyTorch进行管理，具体方法如下：

conda create -n pytorch python=3.6

如果出现如下错误

# 使用清华的镜像链接 conda config --add channels 'https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/' conda config --set show_channel_urls yes

如果还是不行，到打开 C盘/用户/（你的用户名xx）/.condarc，我自己的如下图所示：

把以下这段代码复制到这个.condarc文件中：

channels: - defaults show_channel_urls: true channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda default_channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2 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 simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

保存，即可

查看创建的“房间”

conda info --envs 五、通过“房间”安装PyTorch

激活pytorch房间

conda activate pytorch

当前面从（base）变为（pytorch）时表示此时已经切换到pytorch房间，随后正式进入安装pytorch环节。

安装pytorch

通过pytorch官网,点击Get Started，尴尬没我我们的10.1

将代码粘贴进Anaconda Prompt，将10.2改为10.1

需要安装的东西真的好多啊，漫长的等待啊

如果网络不好有些包无法下载，可以手动安装

conda install pytorch-1.4.0-cpu_py36ha775e86_0.tar.bz2
下载地址：https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/