PyTorch GPU环境安装:windows10+Python3.6+CUDA10.0+PyTorch 0.4.1

话不多说,已安装好anaconda3和CUDA10、cudnn,直接开始安装PyTorch。

创建一个虚拟环境

conda create -n pytorch  python=3.6

PyTorch GPU环境安装:windows10+Python3.6+CUDA10.0+PyTorch 0.4.1_第1张图片

然后进入pytorch官网:https://pytorch.org/

PyTorch GPU环境安装:windows10+Python3.6+CUDA10.0+PyTorch 0.4.1_第2张图片

选择 先前版本 previous versions of PyTorch

PyTorch GPU环境安装:windows10+Python3.6+CUDA10.0+PyTorch 0.4.1_第3张图片

# CUDA 10.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch

如果不能安装可输入一下命令即可:

pip install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-win_amd64.whl

pip install torchvision==0.4.0

PyTorch GPU环境安装:windows10+Python3.6+CUDA10.0+PyTorch 0.4.1_第4张图片

在cmd上看查看是否安装成功?输入以上命令。

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