windows环境下使用anaconda安装pytorch,cuda,cudatoolkit,cuDNN

安装anaconda,以及创建新的环境就不再赘述。

直接从环境配置开始

conda activate your_env后;

1、优先设置一下conda的清华源,下载会很快。

依次输入以下四条指令

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

2、进入pytorch官网,安装torch

选择合适的选项,复制最下面一行到CMD中执行

Start Locally | PyTorch

3、安装cuda,cudatoolkit,cuDNN

首先,输入nvidia-smi查看信息,但我觉得不重要

直接在终端输入以下指令,安装:

conda install cudatoolkit
conda install cudnn

应该会自己适配版本。等待安装完就行了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值