深度学习PyTorch环境安装

一、Anaconda3

1.下载清华源的Anaconda3

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D

2终端运行

bash Anaconda3....(下载的文件名),一路enter

直到出现initialize输入yes

2.配置环境变量

which anaconda #找到安装好的conda路径
vim ~/.bashrc
export PATH="/conda的路径:$PATH"  #例如export PATH="/home/fyq/anaconda3/bin:$PATH"
source ~/.bashrc

3.添加源

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 --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

4.创建虚拟环境(pytorch为虚拟环境名字)

conda create -n pytorch python=3.9

5.进入虚拟环境

conda activate pytorch 

退出命令:conda deactivate pytorch

二、创建pytorch(可以在官网选择需要的版本生成对应命令),此命令为pytorch1.11,

!!!!!!!!!!注意,去掉命令行后面的-c pytorch,不然非常慢!!!

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

检验

进入虚拟环境后

python
import torch
torch.cuda.is_available()

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值