安裝TensorFlow-gpu的注意點

直接pip install tensorflow,安裝到的是tensorflow-gpu 1.14.0,對應的CUDA版本是10.0,我爲啥知道?本以爲對應的是最新的CUDA10.1結果安好就悲劇了,還好軟件提示:無法找到cudart64_100,請下載CUDA10.0,不過後面給的連接比較坑爹,給的是cuda9.0的,老實的我又中招了,下載了cuda9.0還是用不了。。。後來在下面找到了10.0
相關文章
相關標籤/搜索