准备工作:
1.安装VScode
2.安装miniconda
3.安装CUDA9.0
4.安装cuDNN7.0
参考文章:python安装配置(miniconda版)(Win10+VScode +miniconda),CUDA9.0+cuDNN7.0安装配置
-------------------------------------以下是正文----------------------------
1.打开miniconda,创建环境
//pytorch为环境名,3.6为python版本号
conda create -n pytorch python=3.6
2.安装pytorch
添加镜像源,注意一行一行输入
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ //这一个是新增加的
然后安装
conda activate pytorch //激活环境
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 //安装cuda9.0对应的pytorch版本
3.新增环境变量
检查是否安装成功,pip show torch
4.新建python文件,输入代码,并运行
import torch
print(torch.__version__)
print(torch.cuda.is_available())
安装成功。
【备注】如果运行过程报错,大概就是找不到numpy,DLL load failed: 找不到指定的模块。
解决方法:用pip重新安装numpy(若numpy下载过程出现错误,是网速问题)
pip uninstall numpy //卸载numpy
pip3 install numpy==1.14.0 //用pip重新安装numpy