pytorch安装配置(VScode +miniconda+pytorch)

准备工作:

1.安装VScode

2.安装miniconda

3.安装CUDA9.0

4.安装cuDNN7.0

参考文章:python安装配置(miniconda版)(Win10+VScode +miniconda)CUDA9.0+cuDNN7.0安装配置

-------------------------------------以下是正文----------------------------

一、安装pytorch

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

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值