GPU pytorch 1.4.0 cuda 10.1 安装

安装版本:pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1,pytorch官网

第一步:安装 conda 镜像通道

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

第二步:

       官方命令:

# CUDA 10.1
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch

        实际执行: 去除 -c pytorch ,因为该参数指的是指定官方的通道,导致下载的路径不是国内源,因此去除该参数

        canda 执行 命令如下:

conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1

 

最后,验证安装是成功:

import torch

 

安装查考文章:

       PyTorch环境配置及安装

你可能感兴趣的:(环境搭建,pytorch)