Pytorch及其環境踩坑記錄+極簡安裝

今天跑模型的時候發現模型，損失及參數在調用.to(torch.device('cuda'))後，在反向計算梯度時內核會直接卡死，發現是cuda版本過舊，於是進行一系列的踩坑，總結經驗如下： 0，注意這裏都是基於python3.7 1，不必分別安裝cuda和pytorch，這樣及其容易造成版本不匹配問題，我們直接使用anaconda的虛擬環境即可。 2.通過虛擬環境打開terminal 3.調用py

>>阅读原文<<