win10+Anaconda+cuda8.0下安装pytorch-gpu以及opencv3

1.打开anaconda prompt新建一个虚拟环境

conda create -n my_pytorch

进入新建的环境

activate my_pytorch

安装必要的包

conda install -n my_pytorch numpy pyyaml mkl setuptools cmake cffi
conda install -n my_pytorch scipy matplotlib

2.修改conda安装源

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

记事本打开C:\Users***.condarc,修改成如下内容

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
ssl_verify: false
show_channel_urls: true

3.下载CUDA版本对应的pytorch安装包

戳此前往
我这里CUDA8.0,因此选择pytorch-1.0.1-py3.7_cuda80_cudnn7_1.tar.bz2下载
cd到改文件目录下进行安装

conda install -n my_pytorch pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2

4.安装torchvison

注意与pytorch版本对应上
版本号查询

pip install torchvision==0.2.0

5.测试

# CUDA TEST
import torch
import torchvision
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))

6.安装opencv3

为什么版本号为3,但是要import cv2
这里的2表示的是C++为基础,而不是版本号
安装依赖

conda install anaconda-client
conda install -chttps://conda.binstar.org/menpo opencv3

安装指南
我太难了,装环境就装了两天!!!
如果实在无法通过命令行安装,点击这里这里直接下载对应版本的OpenCV,然后pip安装
步骤

7.版本号查询

Windows下 anaconda、python、cuda、pytorch 版本查询命令
win10下CUDA,cudnn版本号查询

你可能感兴趣的:(win10+Anaconda+cuda8.0下安装pytorch-gpu以及opencv3)