支持TensorFlow和pytorch在GPU加速下,pycharm中运行。
软件版本
Windows10
NVIDIA GeForce GTX 1660Ti
CUDA11.0
TensorFlow-gpu2.4.0
pytorch1.7.1
cudnn8.0
python3.8.0
anaconda
pycharm community 2020.3
查看适配版本
CUDA
电脑右下角 NVIDIA设置
左下角系统信息
可以看到支持的最高CUDA版本
这里我们选择11.6及以下的版本都是可以的,为了适配tf和torch版本,本次选择11.0。
注意:请确定您的显卡支持CUDA。