GTX 2080 Windows 10 安装 Pytorch 1.1 GPU版本

更新于2019/5/10,请酌情参考自身条件采纳,前期准备如下:

安装步骤

  1. 安装Anaconda,一路Next,在Advanced Option选项,勾选添加进环境变量,安装完成;
# 修改Conda源到中科大镜像
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
  1. 安装CUDA10,选择一个空路径安装,选择自定义安装,并只勾选CUDA组件;
  2. 解压缩你下载的cuDNN v7.5.1;
  3. 参照官网,将解压出来的cuDNN对应的bin/cudnn64_7.dll, include/cudnn64_7.dll, lib/x64/cudnn.lib三个文件,复制到对应CUDA目录下;y
  4. Alt+R,输入cmd,回车进入命令行模式,输入conda create -n py37_pytorch_gpu pip python=3.7 创建conda环境;
  5. 在命令行下,键入conda activate py37_pytorch_gpu;
  6. PyTorch官网找到并键入对应安装命令,我的是conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
  7. 安装完成后,在py37_pytorch_gpu环境下启动python;
import torch

print(torch.cuda.is_avaliable())
# True
# 大功告成!

10 .当然,如果你习惯在Jupyter Notebook下做分析建模,你需要手动将创建的环境与Jupyter绑定(参考方法

source activate myenv
python -m ipykernel install --user --name myenv --display-name "Python (myenv)"

conda install nb_conda_kernels

    原文作者:周知瑞
    原文地址: https://zhuanlan.zhihu.com/p/65362110
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞