首先自己先装好自己的cuda驱动,我自己装的是9.2的版本,装的方法可自行百度
miniconda 安装
mini conda 是conda的轻量级版本,相比anaconda占用空间小。直接在清华镜像下载安装包 https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/
选择python3.8的最新版本,下载完后直接双击安装,选择自己的安装路径,一路next 就行 中间勾选添加环境变量的选项,这里要注意下如果安装的不是默认路径,在安装完后要设置安装主路径的权限,右键->属性->安全->SYSTEM->确认。不然后面安装torch的时候会出错。
安装完后在win10开始栏哪里找到miniconda的快捷方式
打开输入下面命令,显示当前conda 的虚拟环境的路径
conda env list
base为默认环境,可以使用
conda create --name (虚拟环境的名字) python=(python的版本)
创建新的环境,这里我直接使用默认环境
更换为国内源
在c盘的用户目录下打开 .condarc文件更换下载源为国内源,这里用的清华镜像,也可以选择阿里,中科大,豆瓣的镜像,同时添加了pytorch的镜像地址,删除文件中的default选项
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch//win-64/
show_channel_urls: true
更换pip下载为豆瓣源,在同样的目录下新建一个名为pip的文件夹,里面新建一个pip.ini文件,填入下面信息保存
[global]
timeout = 60
index-url = https://pypi.douban.com/simple
trusted-host = pypi.douban.com
timeout是超时时间,单位是秒
opencv-python 安装
python -m pip install opencv-python
pytorch 安装直接可以用下面命令 注意自己CUDA驱动版本要和Pytorch版本相匹配
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2
安装完后测试安装是否成功