一、安装tensorflow:
最重要的,在官网查看对应版本!!
我安装的是tensorflow-gpu_2.4.0
创建虚拟环境
conda create -n 环境名字 python=3.7
python可根据版本要求选择。
激活环境
conda activate 环境名称
安装tensorflow
pip install -i https://pypi.tuna.tsnghua.edu.cn/simple tensorflow-gpu==2.4.0
测试安装成功
python
import tensorflow as tf
print('GPU:',tf.test.is_gpu_available())
结果返回True,即成功。
二、报错
打开jupyter无法建立连接
这个问题得依情况而定,我在终端查看我的错误是缺少‘cusolver64_10.dll’
解决办法:
dll文件下载地址
参考地址
把下好的文件放到这个目录下就好了。
三、在jupyter使用虚拟环境
同上篇pytorch