因为电脑里安装了多个版本的python,所以这里直接使用Anaconda安装,便于管理包
一、安装visual studio 2013 :
后面要用到vs开发环境
二、安装cuda-10.1:
安装时选择自定义,只安装第一个CUDA选项就好(因为笔记本自带的英伟达驱动比这个新),Visual Studio Integration不要勾选,前面装了vs了,已经有集成环境了。
三、安装Anaconda:
下载安装包(最好去官网根据电脑配置下载最新的,晚上下载速度会快点):官网下载链接
注:安装时一路next就好,路径可以选择默认的或者其它的都可以,到下面这一步时选择下面那个比较好,避免和其它软件发生冲突。
安装完成以后记得添加路径到环境变量,如下图所示:
(Scripts是为了找到conda)
环境变量生效以后可以在命令行输入conda info查看信息:
四、安装pytorch
先去官网查看对应的pytorch版本:
注:直接安装下载不动,所以这里通过清华镜像来安装
添加镜像:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/
安装:
conda install pytorch torchvision cudatoolkit=10.1
安装完成以后可以查看版本:
注:记一下关于镜像的几个命令:
conda config --show #显示镜像源
conda config --remove-key channels #删除conda配置的镜像源