安裝CUDA後安裝PyTorch,pip install torch 太慢【解決辦法】 實測可行torch.cuda.is_available() return True

最近學習深度學習,訓練模型時發現用CPU太慢,恰好電腦有一張NVIDIA的1060顯卡,所以就打算下載NVIDIA的運算平臺CUDA來使用GPU加快訓練了。 我的python版本是3.7,裝完CUDA10.2後,接下來就是安裝pytroch了。由於我安裝的是最新的穩定版,所以使用conda安裝發生了找不到Package的問題,於是我用pip安裝   在命令行輸入 pip install torch
相關文章
相關標籤/搜索