win10anacondacuda11.1安装tensorlfow-gpu环境注:之前写的那篇⽂章虽然安装成功了,但⽤不了gpu,今天从新弄⼀下,让它⽀持gpu
本⽂链接:
前期准备:
在安装 tensorflow-gpu 之前,请确认你已经安装好了 cuda ⼯具包及 cudann 补丁
注:需要注册⼀个英伟达的开发者账号
⼀、安装 anaconda
注1:选择适合⾃⼰的版本下载安装即可
注2:为了不必要的⿇烦,建议先卸载系统已安装的python,然后使⽤anaconda默认环境的python做为系统默认的python
⼆、安装 tensorflow-gpu
安装好 anaconda 后,win+q 搜索 ana ,选择打开 Anaconda Navigator
三、修改 anaconda 源
打开 C盘 > ⽤户 > {⽤户名},例如:C:\Users\Tiac
channels:
- defaults
show_channel_urls: true
default_channels:
- mirrors.tuna.tsinghua.edu/anaconda/pkgs/main
- mirrors.tuna.tsinghua.edu/anaconda/pkgs/r
- mirrors.tuna.tsinghua.edu/anaconda/pkgs/msys2
custom_channels:
conda-forge: mirrors.tuna.tsinghua.edu/anaconda/cloud
msys2: mirrors.tuna.tsinghua.edu/anaconda/cloud
bioconda: mirrors.tuna.tsinghua.edu/anaconda/cloud
menpo: mirrors.tuna.tsinghua.edu/anaconda/cloud
pytorch: mirrors.tuna.tsinghua.edu/anaconda/cloud
simpleitk: mirrors.tuna.tsinghua.edu/anaconda/cloudtensorflow版本选择
四、创建 tensorflow-gpu 环境,并安装 tensorflow-gpu
参考链接:
官⽅教程⾥有⼏种⽅法来安装 tensorflow-gpu,我试了,只有安装 nightly 版本的才⽐较正常(⽀持gpu)conda create -n tf-n-gpu python
conda activate tf-n-gpu
pip install tf-nightly-gpu
五、验证环境是否正常
python
import tensorflow as tf
tf.__version__
注:如图所⽰,tensorflow已安装正常,可以识别到电脑的gpu
六、安装 jupyterlab
参考 jupyter 官⽅⽂档,⽤ conda 命令安装 jupyterlab conda install -c conda-forge jupyterlab
七、运⾏ jupyter-lab 及测试
本⽂链接:
完。

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。