conda\pip 安裝pytorch

最近GPU 上重新安裝pytorch 出現了一些問題,整理一下正確安裝pytorch

1.打開pytorch 官網 https://pytorch.org/

如圖生成自己安裝環境對應的命令 ,如果找不到對應版本,可以直接替換版本號即可,例如在 cuda 9.0 下安裝 torch 

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch -c defaults -c numba/label/dev

測試安裝是否成功 (torch.cuda.is_available()) 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章