Anaconda配置pytorch-gpu虚拟环境的图文教程


Posted in Python onApril 16, 2020

1、更新NVIDIA驱动 

选对应自己显卡的驱动,(选studio版本,不要game版本)驱动链接 

Anaconda配置pytorch-gpu虚拟环境的图文教程

2、添加Anaconda清华镜像

方法一:anaconda命令替换

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

(Mark)换回默认源代码:

conda config --remove-key channels

方法二:替换.condarc

Anaconda配置pytorch-gpu虚拟环境的图文教程

show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
custom_channels:
 conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
 simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

3.创建虚拟环境

创建:
conda create -n 环境名 python=X.X
开启:
activate 环境名
关闭:
conda deactivate
删除:
conda remove -n 环境名 --all
添加包:
conda install -n 环境名 包名
移除包:
conda remove -n 环境名 包名

4. 安装pytorch命令——对应情况自选:命令代码链接(pytorch官网)

1.PIP安装(推荐)

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

Anaconda配置pytorch-gpu虚拟环境的图文教程

这个速度比conda稳定 卡住了按回车好像还可以救回来

Anaconda配置pytorch-gpu虚拟环境的图文教程

2.conda安装(不推荐 老是中断)

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch (要去掉-c pytorch 不然还是默认源)
最终输入命令:
conda install pytorch torchvision cudatoolkit=10.1

Anaconda配置pytorch-gpu虚拟环境的图文教程

网络在各处中断 有时候48%又断了

Anaconda配置pytorch-gpu虚拟环境的图文教程

5.PIP安装完之后测试

import torch
flag = torch.cuda.is_available()
print(flag)
 
ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

结果:被conda命令折腾一下午,终于被pip命令解救了!

Anaconda配置pytorch-gpu虚拟环境的图文教程

总结

到此这篇关于Anaconda配置pytorch-gpu虚拟环境步骤整理的文章就介绍到这了,更多相关Anaconda pytorch-gpu虚拟环境内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
python基于windows平台锁定键盘输入的方法
Mar 05 Python
pygame学习笔记(5):游戏精灵
Apr 15 Python
Python中列表元素转为数字的方法分析
Jun 14 Python
Python环境搭建之OpenCV的步骤方法
Oct 20 Python
python爬虫之线程池和进程池功能与用法详解
Aug 02 Python
pygame实现简易飞机大战
Sep 11 Python
代码实例讲解python3的编码问题
Jul 08 Python
python基于property()函数定义属性
Jan 22 Python
Python 实现敏感目录扫描的示例代码
May 21 Python
Python求区间正整数内所有素数之和的方法实例
Oct 13 Python
Python下使用Trackbar实现绘图板
Oct 27 Python
Python pandas读取CSV文件的注意事项(适合新手)
Jun 20 Python
解决paramiko执行命令超时的问题
Apr 16 #Python
python nohup 实现远程运行不宕机操作
Apr 16 #Python
通过python 执行 nohup 不生效的解决
Apr 16 #Python
在python中使用nohup命令说明
Apr 16 #Python
VS2019+python3.7+opencv4.1+tensorflow1.13配置详解
Apr 16 #Python
python实现程序重启和系统重启方式
Apr 16 #Python
Mac PyCharm中的.gitignore 安装设置教程
Apr 16 #Python
You might like
兼容PHP5的PHP目录管理函数库
2008/07/10 PHP
smarty的保留变量问题
2008/10/23 PHP
php控制linux服务器常用功能 关机 重启 开新站点等
2012/09/05 PHP
PHP登录验证码的实现与使用方法
2016/07/07 PHP
破除网页鼠标右键被禁用的绝招大全
2006/12/27 Javascript
模仿JQuery sortable效果 代码有错但值得看看
2009/11/05 Javascript
javascript之bind使用介绍
2011/10/09 Javascript
JavaScript打印iframe内容示例代码
2013/08/20 Javascript
Javascript基于AJAX回调函数传递参数实例分析
2015/12/15 Javascript
一系列Bootstrap导航条使用方法分享
2016/04/29 Javascript
js表单登陆验证示例
2016/10/19 Javascript
详解Vue 非父子组件通信方法(非Vuex)
2017/05/24 Javascript
JS实现发送短信验证后按钮倒计时功能(防止刷新倒计时失效)
2017/07/07 Javascript
vue.js声明式渲染和条件与循环基础知识
2017/07/31 Javascript
js replace 全局替换的操作方法
2018/06/12 Javascript
详解.vue文件中style标签的几个标识符
2018/07/17 Javascript
Vue-cli3简单使用(图文步骤)
2019/04/30 Javascript
Vue+Koa2+mongoose写一个像素绘板的实现方法
2019/09/10 Javascript
vue实现导航菜单和编辑文本的示例代码
2020/07/04 Javascript
js实现轮播图效果 纯js实现图片自动切换
2020/08/09 Javascript
nuxt 路由、过渡特效、中间件的实现代码
2020/11/06 Javascript
[00:33]DOTA2上海特级锦标赛 CDEC战队宣传片
2016/03/04 DOTA
Python文件右键找不到IDLE打开项解决办法
2015/06/08 Python
python Celery定时任务的示例
2018/03/13 Python
python3实现随机数
2018/06/25 Python
python 实现在一张图中绘制一个小的子图方法
2019/07/07 Python
基于MSELoss()与CrossEntropyLoss()的区别详解
2020/01/02 Python
html5 video标签屏蔽右键视频另存为的js代码
2013/11/12 HTML / CSS
关于圣诞节的广播稿
2014/01/26 职场文书
美国留学经济担保书
2014/05/20 职场文书
老干部工作先进集体事迹材料
2014/05/21 职场文书
在校大学生自我评价范文
2014/09/12 职场文书
客户经理岗位职责大全
2015/04/09 职场文书
党小组鉴定意见
2015/06/02 职场文书
公司行政管理制度范本
2015/08/05 职场文书
小区物业管理2015年度工作总结
2015/10/22 职场文书