前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >[python] python 虚拟环境构建 & GPU环境

[python] python 虚拟环境构建 & GPU环境

作者头像
MachineLP
发布2020-10-29 10:20:34
9880
发布2020-10-29 10:20:34
举报
文章被收录于专栏:小鹏的专栏小鹏的专栏

GPU/python环境配置与验证。

(1)GPU加速型实例安装NVIDIA GPU驱动及CUDA工具包

(2)华为云linux服务器部署TensorFlow-gpu全攻略https://www.cnblogs.com/zxyza/p/10535939.html

(3) Ubuntu安装Anaconda3: https://www.jianshu.com/p/d9fb4e65483c

(4)添加环境变量: vim ~/.bashrc

代码语言:javascript
复制
        export PATH="/root/anaconda3/bin:$PATH"
        export PATH=/usr/local/cuda/bin{PATH:+:{PATH}}
        export LD_LIBRARY_PATH=/usr/local/cuda/lib64{LD_LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
        export CUDA_HOME=/usr/local/cuda

(5)source ~/.bashrc

(6)创建虚拟环境:

conda create -n py37 python=3.7

进入环境

source activate py37

conda activate py37

退出环境

source deactivate

conda deactivate

(7)source activate py37

(8)安装tensorflow-gpu:pip install tensorflow-gpu==1.13.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

(9)测试:

import tensorflow as tf

import os

# os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

print('GPU>>>>>>', tf.test.is_gpu_available())

a = tf.constant(2.0)

b = tf.constant(4.0)

print(a + b)

(10) 结果:

GPU>>>>>> True

Tensor("add:0", shape=(), dtype=float32)

(11) 不同版本torch安装:

代码语言:javascript
复制
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0

conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.0

conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1

上述命令直接安装太太太慢了,可以通过更换conda源来加速下载。

代码语言:javascript
复制
# 修改conda配置
vim .condarc

# 在配置钟添加清华源
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - 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/cloud/conda-forge
  - default
show_channel_urls: true


# 安装pytorch和对应版本的cudatoolkit
conda install pytorch=1.4.0 torchvision cudatoolkit=10.1
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-06-24 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档