windows10 安装 pytorch gpu版本
发布时间
阅读量：
阅读量
在PyTorch官方网站上仅限于提供了Linux和Mac操作系统版本的软件包。并未提供Windows系统的软件包。然而Windows系统可以通过其他途径使用PyTorch
首先要安装anaconda：
请访问清华镜像资源库中的Anaconda官方源站点：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/。建议将下载链接拖动至页面底部，并选择最新的版本进行安装。
安装完毕要安装pytorch运行环境：
# If your main Python version is not 3.5 or 3.6
conda create -n test python=3.6 numpy pyyaml mkl然后是安装gpu版本，pytorch：
# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch cuda80由于TensorFlow仅支持CUDA8，并且我已经按照要求完成了安装，请问您是否也遇到了类似的问题？
安装Visual C++ 2015 64 bit
https://www.microsoft.com/zh-cn/download/details.aspx?id=48145 安装这些程序通常是不必要的，这些程序一般都会被安装。
下载安装Cuda Toolkit 8.0
https://developer.nvidia.com/cuda-downloads
因为TensorFlow只能用 CUDA8，我装的8.0，不过听说pytorch支持9.0，可以尝试。
下载8.0:
选择版本：
两个都要下载：
下载完毕安装就行了。
安装cuDNN
官网维护了，度盘链接在此：
https://pan.baidu.com/s/1nwAYgIl
解压后覆盖到Cuda Toolkit 8.0安装目录中，这样就可以使用了。
测试一下：
打开python，输入如下代码：
import torch
torch.cuda.is_available()返回True证明成功了。
结束。。。。。。。。。。。。。。。。。。。。。。
全部评论 (0)
还没有任何评论哟~