Anaconda3安裝Pytorch

1檢查cuda版本 pytorch版本需要跟cuda對應 輸入命令:cat /usr/local/cuda/version.txt 我的cuda是9.0 2.pytorch官網下載 根據cuda版本,pytorch官網,下載對應的版本。複製最後一行語句。 3安裝 輸入命令conda install pytorch==1.1.0 torchvision ==0.3.0 cudatoolkit=9.0
相關文章
相關標籤/搜索