(1)安装anaconda(注意选择的时候选择anaconda3)
(2)打开anaconda prompt
·设置清华源
首先生成 conda config
去c盘用户里找到 .condarc文件,用记事本打开,将
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- defaults
show_channel_urls: true
复制进去并保存。
·创建虚拟环境:conda create -n tensorflow2.4.0 python=3.7
·激活环境: conda activate tensorflow2.4.0
输入命令:conda search cudatoolkit --info寻找可安装的版本
我选择的是11.0
conda install cudatoolkit==11.0.221
之后输入命令:conda search cudnn --info 找对应的cudnn
conda install cudnn==8.9.2.26
最后安装tensorflow-gpu
pip install tensorflow-gpu==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple --timeout=120
成功!接下来输入python
输入 import tensorflow as tf
print(tf._version_)
import tensorflow.compat.v1 as tf
>>>
... sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
显示有gpu了!!!
在pycharm里将这个虚拟环境导入就可以了:settings,解释器,选择conda enviroment,选择已存在的环境。之后按各种模块即可。