pyTorch学习1:环境配置

1、添加国内镜像
参考:conda安装Pytorch下载过慢解决办法

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

2、安装pytorch
进入pytorch官网选择操作系统和版本,下面就能给出推荐的安装命令,建议使用anaconda安装

# 注意不要按照官网的提示在最后加 -c pytorch,去掉后才会从添加的镜像源下载
conda install pytorch torchvision cuda80
# 如果需要安装指定版本的pytorch,参照:
# https://pytorch.org/get-started/previous-versions/ 例如:
conda install pytorch=0.4.0 torchvision cuda80

3、验证是否安装成功

source activate your_conda_env # 或者是conda activate,谁能告诉我二者的区别。。。
python
>>>import torch
>>>print(torch.__version__)

若安装成功,会输出当前pytorch的版本号
根据笔者经验,大多安装不成功的原因均来自cuda版本,设置好安装的cudatoolkit版本即可

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值