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中threading模块join函数用法实例分析
Jun 04 Python
python3.5实现socket通讯示例(TCP)
Feb 07 Python
TensorFlow平台下Python实现神经网络
Mar 10 Python
Python中pip更新和三方插件安装说明
Jul 08 Python
Ubuntu下升级 python3.7.1流程备忘(推荐)
Dec 10 Python
WxPython建立批量录入框窗口
Feb 27 Python
python logging模块书写日志以及日志分割详解
Jul 22 Python
python运用pygame库实现双人弹球小游戏
Nov 25 Python
Python 日期的转换及计算的具体使用详解
Jan 16 Python
DataFrame 数据合并实现(merge,join,concat)
Jun 14 Python
利用Python优雅的登录校园网
Oct 21 Python
关于python类SortedList详解
Sep 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
使用PHP实现二分查找算法代码分享
2011/06/24 PHP
基于PHP5魔术常量与魔术方法的详解
2013/06/13 PHP
php 字符串中的\n换行符无效、不能换行的解决方法
2014/04/02 PHP
js 判断浏览器类型 去全角、半角空格 自动关闭当前窗口
2009/04/10 Javascript
jquery异步请求实例代码
2011/06/21 Javascript
深入理解JavaScript系列(6) 强大的原型和原型链
2012/01/15 Javascript
jQuery $.data()方法使用注意细节
2012/12/31 Javascript
jQuery表单域属性过滤器用法分析
2015/02/10 Javascript
JavaScript实现判断图片是否加载完成的3种方法整理
2015/03/13 Javascript
详解JS中Array对象扩展与String对象扩展
2016/01/07 Javascript
JS表格组件BootstrapTable行内编辑解决方案x-editable
2016/09/01 Javascript
微信小程序中使用Promise进行异步流程处理的实例详解
2017/08/17 Javascript
vue页面跳转后返回原页面初始位置方法
2018/02/11 Javascript
JS实现页面跳转与刷新的方法汇总
2019/08/30 Javascript
使用xampp将angular项目运行在web服务器的教程
2019/09/16 Javascript
three.js着色器材质的内置变量示例详解
2020/08/16 Javascript
js在HTML的三种引用方式详解
2020/08/29 Javascript
Vue SPA 首屏优化方案
2021/02/26 Vue.js
python实现代理服务功能实例
2013/11/15 Python
Tensorflow 查看变量的值方法
2018/06/14 Python
python将pandas datarame保存为txt文件的实例
2019/02/12 Python
Django组件cookie与session的具体使用
2019/06/05 Python
Python3环境安装Scrapy爬虫框架过程及常见错误
2019/07/12 Python
python pillow模块使用方法详解
2019/08/30 Python
Python列表解析操作实例总结
2020/02/26 Python
浅谈matplotlib.pyplot与axes的关系
2020/03/06 Python
使用分层画布来优化HTML5渲染的教程
2015/05/08 HTML / CSS
皇马官方商城:Real Madrid Store
2016/09/02 全球购物
Banggood官网:面向全球客户的综合商城
2017/04/19 全球购物
西班牙多品牌鞋店连锁店:Krack
2018/11/30 全球购物
优质服务活动实施方案
2014/05/02 职场文书
本科毕业生自荐信
2014/05/26 职场文书
缓刑期间思想汇报范文
2014/10/10 职场文书
幼儿园老师工作总结2015
2015/05/22 职场文书
2015年校本培训工作总结
2015/07/24 职场文书
Java elasticsearch安装以及部署教程
2021/06/28 Java/Android