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版本号查询