安裝pytorch anaconda
- IT科技
- 關注:2.57W次
1、設定虛擬環境。之所以要設定虛擬環境,是因為這個 python裡面的包有很多個,有些個包可以會存在相互干擾的問題,所以用這個虛擬環境來把他們來隔絕開。
建立虛擬環境:conda create -n torch python=3.6
啟用虛擬環境:activate torch,
當啟用torch環境之後,左邊的環境就由base變為了torch。
2、去Pytorch的官網(https://pytorch.org/get-started/locally/)下找到PyTorch安裝需要使用的命令 。
將程式碼conda install pytorch torchvision cudatoolkit=10.1 -c pytorch粘到命令列裡面。
進入python環境,然後輸入import torch 和import torchvision驗證,如果沒有報錯的話,就說明安裝已經成功了。
- 文章版權屬於文章作者所有,轉載請註明 https://shqsg.com/dianzi/69rrm3.html