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脚本帮你找出微信上删除你好友的人
Jan 07 Python
Python实现判断一个字符串是否包含子串的方法总结
Nov 21 Python
PyQt实现界面翻转切换效果
Apr 20 Python
Django框架的使用教程路由请求响应的方法
Jul 03 Python
对python3中pathlib库的Path类的使用详解
Oct 14 Python
对Python生成汉字字库文字,以及转换为文字图片的实例详解
Jan 29 Python
Python3.5面向对象程序设计之类的继承和多态详解
Apr 24 Python
Python基于gevent实现高并发代码实例
May 15 Python
python 使用递归的方式实现语义图片分割功能
Jul 16 Python
Python日志打印里logging.getLogger源码分析详解
Jan 17 Python
使用Python下载抖音各大V视频的思路详解
Feb 06 Python
用Python编写简单的gRPC服务的详细过程
Jul 04 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
IIS6.0 开启Gzip方法及PHP Gzip函数分享
2014/06/08 PHP
在Mac OS的PHP环境下安装配置MemCache的全过程解析
2016/02/15 PHP
PHP7 新特性详细介绍
2016/09/06 PHP
Yii-自定义删除确认弹框(zyd)jquery实现代码
2013/03/04 Javascript
分享10个原生JavaScript技巧
2015/04/20 Javascript
js实现a标签超链接提交form表单的方法
2015/06/24 Javascript
第五章之BootStrap 栅格系统
2016/04/25 Javascript
总结Javascript中的隐式类型转换
2016/08/24 Javascript
详解堆的javascript实现方法
2016/11/29 Javascript
浅谈angular2路由预加载策略
2017/10/04 Javascript
使用 Vue 绑定单个或多个 Class 名的实例代码
2018/01/08 Javascript
zTree 树插件实现全国五级地区点击后加载的示例
2018/02/05 Javascript
vue父组件点击触发子组件事件的实例讲解
2018/02/08 Javascript
vue里面v-bind和Props 利用props绑定动态数据的方法
2018/08/27 Javascript
基于jQuery ztree实现表格风格的树状结构
2018/08/31 jQuery
微信小程序实现展示评分结果功能
2019/02/15 Javascript
使用JavaScript解析URL的方法示例
2019/03/01 Javascript
js实现unicode码字符串与utf8字节数据互转详解
2019/03/21 Javascript
JavaScript函数式编程(Functional Programming)高阶函数(Higher order functions)用法分析
2019/05/22 Javascript
express + jwt + postMan验证实现持久化登录
2019/06/05 Javascript
vue父子组件通信的高级用法示例
2019/08/29 Javascript
vuex vue简单使用知识点总结
2019/08/29 Javascript
python绘制铅球的运行轨迹代码分享
2017/11/14 Python
对DataFrame数据中的重复行,利用groupby累加合并的方法详解
2019/01/30 Python
python 按钮点击关闭窗口的实现
2020/03/04 Python
打印tensorflow恢复模型中所有变量与操作节点方式
2020/05/26 Python
BeautifulSoup获取指定class样式的div的实现
2020/12/07 Python
AmazeUI的下载配置与Helloworld的实现
2020/08/19 HTML / CSS
纽约21世纪百货官网:Century 21
2016/08/27 全球购物
土木工程建筑专业毕业生求职信
2013/10/21 职场文书
幼教毕业生自我鉴定
2014/01/12 职场文书
预备党员综合考察材料
2014/05/31 职场文书
党小组评议意见
2015/06/02 职场文书
培训计划通知
2015/07/15 职场文书
2015年文秘个人工作总结
2015/10/14 职场文书
Java中Dijkstra(迪杰斯特拉)算法
2022/05/20 Java/Android