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比较两个图片相似度的方法
Mar 13 Python
21行Python代码实现拼写检查器
Jan 25 Python
python处理两种分隔符的数据集方法
Dec 12 Python
Django 使用easy_thumbnails压缩上传的图片方法
Jul 26 Python
Python如何使用Gitlab API实现批量的合并分支
Nov 27 Python
python实现tail -f 功能
Jan 17 Python
解决Pycharm中恢复被exclude的项目问题(pycharm source root)
Feb 14 Python
浅谈ROC曲线的最佳阈值如何选取
Feb 28 Python
python字典和json.dumps()的遇到的坑分析
Mar 11 Python
使用 prometheus python 库编写自定义指标的方法(完整代码)
Jun 29 Python
Django Form常用功能及代码示例
Oct 13 Python
python中requests库+xpath+lxml简单使用
Apr 29 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的远程图片抓取函数分享
2013/09/25 PHP
合并ThinkPHP配置文件以消除代码冗余的实现方法
2014/07/22 PHP
Yii2创建控制器(createController)方法详解
2016/07/23 PHP
php 如何禁用eval() 函数实例详解
2016/12/01 PHP
window.parent调用父框架时 ie跟火狐不兼容问题
2009/07/30 Javascript
鼠标事件延时切换插件
2011/03/12 Javascript
js获得当前时区夏令时发生和终止的时间代码
2014/02/23 Javascript
Node.js模块封装及使用方法
2016/03/06 Javascript
Node.js+Express配置入门教程详解
2016/05/19 Javascript
js发送短信倒计时的简单实现方法
2016/09/08 Javascript
详解vuelidate 对于vueJs2.0的验证解决方案
2017/03/09 Javascript
nodejs+express实现文件上传下载管理网站
2017/03/15 NodeJs
微信小程序实战之仿android fragment可滑动底部导航栏(4)
2020/04/16 Javascript
js轮播图的插件化封装详解
2017/07/17 Javascript
解决layer 关闭当前弹窗 关闭遮罩层 input值获取不到的问题
2019/09/25 Javascript
微信小程序个人中心的列表控件实现代码
2020/04/26 Javascript
js实现无缝轮播图特效
2020/05/09 Javascript
Vue 自适应高度表格的实现方法
2020/05/13 Javascript
ES6函数和数组用法实例分析
2020/05/23 Javascript
JS中队列和双端队列实现及应用详解
2020/09/29 Javascript
Python每天必学之bytes字节
2016/01/28 Python
Django 创建后台,配置sqlite3教程
2019/11/18 Python
安装PyInstaller失败问题解决
2019/12/14 Python
美国男士西装打折店:Jos. A. Bank
2017/11/13 全球购物
Lookfantastic葡萄牙官方网站:欧洲第一大化妆品零售商
2018/03/17 全球购物
英国奢侈皮具品牌:Aspinal of London
2018/09/02 全球购物
Theflamel意大利:女士奢华服装、鞋子和配件
2020/01/11 全球购物
数控技术专业毕业自荐书范文
2014/02/05 职场文书
食堂采购员岗位职责
2014/03/17 职场文书
家长会演讲稿
2014/04/26 职场文书
信用社竞聘演讲稿
2014/05/16 职场文书
室内设计专业自荐信
2014/05/31 职场文书
四风对照检查材料思想汇报
2014/09/20 职场文书
2019邀请函格式及范文
2019/05/20 职场文书
python flask框架快速入门
2021/05/14 Python
springboot 多数据源配置不生效遇到的坑及解决
2021/11/17 Java/Android