pytorch GPU版本安装 win10

我原先装的是pytorch cpu版本，在复现论文源码的时候提示AssertionError: Torch not compiled with CUDA enabled，查了一下资料，又看了一下电脑的pytorch版本，才发现原来装的是cpu版本，应该装成gpu版本。在win10环境下安装pytorch gpu版本的步骤如下。
首先在cmd或者anaconda prompt中创建名为pytorch_gpu的环境，制定python版本，我用的是python3.7

conda create -n pytorch_gpu pip python=3.7

继续安装：

Collecting package metadata (repodata.json): done
Solving environment: done## Package Plan ##environment location: D:\Anaconda\envs\pytorch_gpuadded / updated specs:- pip- python=3.7The following packages will be downloaded:package                    |            build---------------------------|-----------------python-3.7.6               |       h60c2a47_2        18.3 MB  defaultssetuptools-44.0.0          |           py37_0         671 KB  defaultsvc-14.1                    |       h21ff451_1           4 KB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123vs2017_runtime-15.4.27004.2010|                1         1.6 MB  https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123------------------------------------------------------------Total:        20.6 MBThe following NEW packages will be INSTALLED:ca-certificates    anaconda/pkgs/main/win-64::ca-certificates-2019.11.27-0certifi            anaconda/pkgs/main/win-64::certifi-2019.11.28-py37_0openssl            anaconda/pkgs/main/win-64::openssl-1.1.1d-he774522_3pip                anaconda/pkgs/main/win-64::pip-19.3.1-py37_0python             anaconda/pkgs/main/win-64::python-3.7.6-h60c2a47_2setuptools         anaconda/pkgs/main/win-64::setuptools-44.0.0-py37_0sqlite             anaconda/pkgs/main/win-64::sqlite-3.30.1-he774522_0vc                 anaconda/cloud/peterjc123/win-64::vc-14.1-h21ff451_1vs2015_runtime     anaconda/pkgs/main/win-64::vs2015_runtime-14.16.27012-hf0eaf9b_1vs2017_runtime     anaconda/cloud/peterjc123/win-64::vs2017_runtime-15.4.27004.2010-1wheel              anaconda/pkgs/main/win-64::wheel-0.33.6-py37_0wincertstore       anaconda/pkgs/main/win-64::wincertstore-0.2-py37_0Proceed ([y]/n)? y
# 忽略中间一顿操作
...done
Executing transaction: done
#
# To activate this environment, use
#
#     $ conda activate pytorch_gpu
#
# To deactivate an active environment, use
#
#     $ conda deactivate

激活pytorch_gpu环境，

conda activate pytorch_gpu

按照pytorch官网上的提示，我们选择合适版本进行安装，我用的是cuda10.0和python3.7，其中cuda版本根据nvidia control panel（nvidia控制面板里面的系统信息里面有提示）具体而定。安装指令：

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

但是安装的时候提示以下问题：

Collecting package metadata (repodata.json): done
Solving environment: done## Package Plan ##environment location: D:\Anaconda\envs\pytorch_gpuadded / updated specs:- cudatoolkit=10.0- pytorch- torchvisionThe following packages will be downloaded:package                    |            build---------------------------|-----------------freetype-2.9.1             |       ha9979f8_1         470 KB  defaultsmkl-service-2.3.0          |   py37hb782905_0         200 KB  defaultsmkl_fft-1.0.15             |   py37h14836fe_0         137 KB  defaultsmkl_random-1.1.0           |   py37h675688f_0         270 KB  defaultsnumpy-1.17.4               |   py37h4320e6b_0           5 KB  defaultsnumpy-base-1.17.4          |   py37hc3f5095_0         4.8 MB  defaultsolefile-0.46               |           py37_0          49 KB  defaultspillow-7.0.0               |   py37hcc1f983_0         726 KB  defaultspytorch-1.2.0              |py3.7_cuda100_cudnn7_1       479.3 MB  pytorchtk-8.6.8                   |       hfa6e2cd_0         3.8 MB  defaultstorchvision-0.4.0          |       py37_cu100         2.3 MB  pytorch------------------------------------------------------------Total:       492.0 MBThe following NEW packages will be INSTALLED:blas               anaconda/pkgs/main/win-64::blas-1.0-mklcffi               anaconda/pkgs/main/win-64::cffi-1.13.2-py37h7a1dbc1_0cudatoolkit        anaconda/pkgs/main/win-64::cudatoolkit-10.0.130-0freetype           anaconda/pkgs/main/win-64::freetype-2.9.1-ha9979f8_1icc_rt             anaconda/pkgs/main/win-64::icc_rt-2019.0.0-h0cc432a_1intel-openmp       anaconda/pkgs/main/win-64::intel-openmp-2019.4-245jpeg               anaconda/pkgs/main/win-64::jpeg-9b-hb83a4c4_2libpng             anaconda/pkgs/main/win-64::libpng-1.6.37-h2a8f88b_0libtiff            anaconda/pkgs/main/win-64::libtiff-4.1.0-h56a325e_0mkl                anaconda/pkgs/main/win-64::mkl-2019.4-245mkl-service        anaconda/pkgs/main/win-64::mkl-service-2.3.0-py37hb782905_0mkl_fft            anaconda/pkgs/main/win-64::mkl_fft-1.0.15-py37h14836fe_0mkl_random         anaconda/pkgs/main/win-64::mkl_random-1.1.0-py37h675688f_0ninja              anaconda/pkgs/main/win-64::ninja-1.9.0-py37h74a9793_0numpy              anaconda/pkgs/main/win-64::numpy-1.17.4-py37h4320e6b_0numpy-base         anaconda/pkgs/main/win-64::numpy-base-1.17.4-py37hc3f5095_0olefile            anaconda/pkgs/main/win-64::olefile-0.46-py37_0pillow             anaconda/pkgs/main/win-64::pillow-7.0.0-py37hcc1f983_0pycparser          anaconda/pkgs/main/win-64::pycparser-2.19-py37_0pytorch            pytorch/win-64::pytorch-1.2.0-py3.7_cuda100_cudnn7_1six                anaconda/pkgs/main/win-64::six-1.13.0-py37_0tk                 anaconda/pkgs/main/win-64::tk-8.6.8-hfa6e2cd_0torchvision        pytorch/win-64::torchvision-0.4.0-py37_cu100xz                 anaconda/pkgs/main/win-64::xz-5.2.4-h2fa13f4_4zlib               anaconda/pkgs/main/win-64::zlib-1.2.11-h62dcd97_3zstd               anaconda/pkgs/main/win-64::zstd-1.3.7-h508b16e_0Proceed ([y]/n)? yDownloading and Extracting Packages
mkl_fft-1.0.15       | 137 KB    | ############################################################################ | 100%
numpy-1.17.4         | 5 KB      | ############################################################################ | 100%
mkl-service-2.3.0    | 200 KB    | ############################################################################ | 100%
torchvision-0.4.0    | 2.3 MB    | ############################################################################ | 100%
pillow-7.0.0         | 726 KB    | ############################################################################ | 100%
pytorch-1.2.0        | 479.3 MB  | #############################################################1               |  80%
numpy-base-1.17.4    | 4.8 MB    |                                                                              |   0%
freetype-2.9.1       | 470 KB    |                                                                              |   0%
olefile-0.46         | 49 KB     | ############################################################################ | 100%
mkl_random-1.1.0     | 270 KB    | ############################################################################ | 100%
tk-8.6.8             | 3.8 MB    | ############################################################################ | 100%CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2>
Elapsed: -An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.ProxyError: Conda cannot proceed due to an error in your proxy configuration.
Check for typos and other configuration errors in any '.netrc' file in your home directory,
any environment variables ending in '_PROXY', and any other system-wide proxy
configuration settings.ProxyError: Conda cannot proceed due to an error in your proxy configuration.
Check for typos and other configuration errors in any '.netrc' file in your home directory,
any environment variables ending in '_PROXY', and any other system-wide proxy
configuration settings.

这是因为conda install pytorch torchvision cudatoolkit=10.1 -c pytorch，是默认到pytorch官网上下载的，下载特别慢，而且容易出现中途连接不上网页、中断下载的情况。为了能够顺利地安装，不出现中断、提升下载速度，要到其他的镜像网址进行下载。清华镜像网址从2019年开始不再更新维护了，我们选择中科大的网址，在命令行中输入以下指令：

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

可以看到，能够顺利进行安装

Collecting package metadata (repodata.json): done
Solving environment: done## Package Plan ##environment location: D:\Anaconda\envs\pytorch_gpuadded / updated specs:- cudatoolkit=10.0- pytorch- torchvisionThe following packages will be downloaded:package                    |            build---------------------------|-----------------freetype-2.9.1             |       ha9979f8_1         470 KB  defaultsnumpy-base-1.17.4          |   py37hc3f5095_0         4.8 MB  defaultspytorch-1.2.0              |py3.7_cuda100_cudnn7_1       479.3 MB  pytorch------------------------------------------------------------Total:       484.5 MBThe following NEW packages will be INSTALLED:blas               anaconda/pkgs/main/win-64::blas-1.0-mklcffi               anaconda/pkgs/main/win-64::cffi-1.13.2-py37h7a1dbc1_0cudatoolkit        anaconda/pkgs/main/win-64::cudatoolkit-10.0.130-0freetype           anaconda/pkgs/main/win-64::freetype-2.9.1-ha9979f8_1icc_rt             anaconda/pkgs/main/win-64::icc_rt-2019.0.0-h0cc432a_1intel-openmp       anaconda/pkgs/main/win-64::intel-openmp-2019.4-245jpeg               anaconda/pkgs/main/win-64::jpeg-9b-hb83a4c4_2libpng             anaconda/pkgs/main/win-64::libpng-1.6.37-h2a8f88b_0libtiff            anaconda/pkgs/main/win-64::libtiff-4.1.0-h56a325e_0mkl                anaconda/pkgs/main/win-64::mkl-2019.4-245mkl-service        anaconda/pkgs/main/win-64::mkl-service-2.3.0-py37hb782905_0mkl_fft            anaconda/pkgs/main/win-64::mkl_fft-1.0.15-py37h14836fe_0mkl_random         anaconda/pkgs/main/win-64::mkl_random-1.1.0-py37h675688f_0ninja              anaconda/pkgs/main/win-64::ninja-1.9.0-py37h74a9793_0numpy              anaconda/pkgs/main/win-64::numpy-1.17.4-py37h4320e6b_0numpy-base         anaconda/pkgs/main/win-64::numpy-base-1.17.4-py37hc3f5095_0olefile            anaconda/pkgs/main/win-64::olefile-0.46-py37_0pillow             anaconda/pkgs/main/win-64::pillow-7.0.0-py37hcc1f983_0pycparser          anaconda/pkgs/main/win-64::pycparser-2.19-py37_0pytorch            pytorch/win-64::pytorch-1.2.0-py3.7_cuda100_cudnn7_1six                anaconda/pkgs/main/win-64::six-1.13.0-py37_0tk                 anaconda/pkgs/main/win-64::tk-8.6.8-hfa6e2cd_0torchvision        pytorch/win-64::torchvision-0.4.0-py37_cu100xz                 anaconda/pkgs/main/win-64::xz-5.2.4-h2fa13f4_4zlib               anaconda/pkgs/main/win-64::zlib-1.2.11-h62dcd97_3zstd               anaconda/pkgs/main/win-64::zstd-1.3.7-h508b16e_0Proceed ([y]/n)? yDownloading and Extracting Packages
numpy-base-1.17.4    | 4.8 MB    | ############################################################################ | 100%
pytorch-1.2.0        | 479.3 MB  | ############################################################################ | 100%
freetype-2.9.1       | 470 KB    | ############################################################################ | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done

pytorch GPU版本安装 win10相关推荐

pytorch gpu版本安装
1,安装anaconda 2,查看显卡版本信息 3,安装对应的驱动程序cuda tooltik 4,安装pytorch 首先安装anaconda ,安装anaconda的目的在于创建虚拟环境,因为有的 ...
PyTorch 学习by_LX（一）：pytorch GPU版本安装等碎碎念
新买的RTX2070到了,一顿操作后装的是cuda11.0,而pytorch目前支持的最高cuda版本为10.2:无奈需要重装,本想在不卸载cuda11.0的条件下安装10.2,一顿操作猛如虎,结果运 ...
关于pytorch GPU版本安装的问题
在官网使用 conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia 命令安装结果是CPU ...
pytorch，torch，torchvision的gpu版本安装避坑
pytorch,torch,torchvision的gpu版本安装避坑在线or离线安装 pytorch torch和torchvision 在线or离线安装最近使用pytorch跑深度学习模型,需 ...
Pytorch GPU版本whl文件安装
Pytorch GPU版本whl文件安装安装pytorch的时候,用pip安装时网速实在太慢,换源也不太行,1.2G的文件,一个网络波动就开始疯狂红字.因此使用whl文件进行安装用whl文件进行安 ...
Pytorch：入门指南和 PyTorch 的 GPU版本安装(非常详细)
Pytorch: 入门指南和 PyTorch 的 GPU版本安装(非常详细) Copyright: Jingmin Wei, Pattern Recognition and Intelligent S ...
TensorFlow GPU 版本安装个人总结：Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0
TensorFlow GPU 版本安装个人总结:Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0 接触机器学 ...
【深度学习】PyTorch 历史版本安装-祖传老代码运行刚需
最新 PyTorch 安装以及 CUDA 版本如果要安装最新的 PyTorch 其实是很简单的,直接到官网首页就有各种系统的安装方法,如同所示: 这里有一个是 CUDA 版本,实测需要一模一样,这 ...
pytorch gpu版本的最全安装教程，含环境配置、CUDA（百度云下载）安装程序失败解决方案
看过我cpu版本的pytorch小白安装教程CPU版pytorch小白教程应该已经会设置清华源了,这里不再叙述,gpu版本的pytorch会默认以清华镜像源来下载,速度不用担心. 请关闭杀毒软件尤其 ...

pytorch GPU版本安装 win10

pytorch GPU版本安装 win10相关推荐

最新文章

热门文章