Ubuntu tensorflow2.X 调用cuda10.2
已经安装了CUDA 10.2,然后才发现TensorFlow 2.2只支持CUDA 10.1,CUDA10.2没有对应的tensorflow版本。采取下面的办法建一个软链接即可。
注意,我的cuda安装在conda虚拟环境里,不使用根目录里安装的cuda
ln -s ~/anaconda3/envs/pytorch/lib/libcudart.so.10.2.89 ~/anaconda3/envs/pytorch/lib/libcudart.so.10.1
相关:conda创建虚拟环境 和
原创
2022-04-11 10:19:57 ·
1004 阅读 ·
0 评论