Win10 Anaconda pytorch GPU

电脑当前环境

win10 64bit

cuda90 (需要提前安装好)

注意⚠️:若你的cuda是其它版本,那么后面的pytorch也要装相应的cuda版本

 

镜像下载安装（强烈推荐）

首先进入此链接 torch清华镜像源链接.

 

分别下载：

pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2

torchvision-0.3.0-py37_cu90_1.tar.bz2

 

执行指令：

 

创建一个conda的环境,我这里是brats

conda create -n brats python=3.7

进入新环境brats,并安装一些依赖项

conda activate brats

conda install numpy mkl cffi

分别执行一下命令进行安装

conda install --offline .\pytorch-1.1.0-py3.7_cuda90_cudnn7_1.tar.bz2

conda install --offline .\torchvision-0.3.0-py37_cu90_1.tar.bz2

 

再安装个依赖项

conda install Pillow=6.1

 

检验

在命令行窗口依次输入

python

import torch

import torchvision

print(torch.__version__)

print(torchvision.__version__)

 

 

 

 

整理不易

点赞收藏关注喔~

来源：CSDN

作者：走上七彩虹

链接：https://blog.csdn.net/weixin_40519315/article/details/104142716