基於anaconda和python3.7安裝pytorch1.5.0

一開始看到自己電腦的CUDA版本爲11.0,以爲不能支持,實際上在虛擬環境中可以安裝cudatoolkit的任意版本: 1.打開anaconda prompt,建立一個新環境 conda create -n your_env_name python=3.7 2.安裝cudatoolkit和對應的cudnn conda install cudatoolkit=10.2 conda install c
相關文章
相關標籤/搜索