anaconda虚拟环境下镜像安装PyTorch,利用jupyter notebook访问虚拟环境

创建虚拟环境并镜像安装PyTorch

1、conda create -n torch2 python=3.7
torch2为我们创建的虚拟环境名称,用户自定义名称。
2、conda activate torch2
切换到虚拟环境
如图所示:
在这里插入图片描述
我们可以看到(torch2),说明已经切换成功。
3、更换镜像源(注意顺序)
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
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
4、运行执行安装pytorch
在这里插入图片描述
pytorch提供的下载命令为:conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
但是,运行此命令未能成功安装。
运行:conda install pytorch torchvision cudatoolkit=10.1 成功安装(亲测有效)
cuda选择的是10.1

利用jupyter notebook访问虚拟环境

在base虚拟环境下输入以下命令:
1、安装nb_conda:
conda install nb_conda
2、安装ipykernel
conda install ipykernel
3、为虚拟环境创建kernel文件:
conda install -n torch2 ipykernel

如图:
在这里插入图片描述
出现了Python[conda env:torch2]
在创建新的python文件时选中Python[conda env:torch2]即可。

下面我们在Jupyter Notebook测试PyTorch GPU版本安装是否成功:

import torch

print(torch.__version__)
print(torch.cuda.is_available())

输出结果:
1.4.0
True
Pytorch版本为1.4.0
GPU版本安装成功。

  • 5
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值