前言
不想用tensorflow,但是有的代码,没有torch版本的,只能捡起来tensorflow。
参考
https://blog.csdn.net/m0_68233404/article/details/128525860
电脑配置
项目 | Value |
---|---|
OS | Windows |
GPU | RTX3060 |
CUDA | 11.2 |
python | 3.8 |
Anaconda | 参考 |
Anconda构建python=3.8虚拟环境
# 创建
conda create --prefix=tensorflowGPUEnv python=3.8
# 激活
conda activate tensorflowGPUEnv的绝对路径
安装requirements
在requirements中的tensorflow版本与自己电脑cuda版本匹配的情况下,可以使用
pip install -r requirements的绝对路径 -i https://pypi.tuna.tsinghua.edu.cn/simple