關閉→
當前位置:知科普>IT科技>anaconda如何安裝pytorch

anaconda如何安裝pytorch

知科普 人氣:2.46W

1、設置虛擬環境。之所以要設置虛擬環境,是因為這個 python裏面的包有很多個,有些個包可以會存在相互干擾的問題,所以用這個虛擬環境來把他們來隔絕開。

創建虛擬環境:conda create -n torch python=3.6

anaconda如何安裝pytorch

激活虛擬環境:activate torch,

anaconda如何安裝pytorch 第2張

當激活torch環境之後,左邊的環境就由base變為了torch。

2、去Pytorch的官網(https://pytorch.org/get-started/locally/)下找到PyTorch安裝需要使用的命令 。

anaconda如何安裝pytorch 第3張

將代碼conda install pytorch torchvision cudatoolkit=10.1 -c pytorch粘到命令行裏面。

anaconda如何安裝pytorch 第4張

進入python環境,然後輸入import torch 和import torchvision驗證,如果沒有報錯的話,就説明安裝已經成功了。

TAG標籤:#anaconda #pytorch #安裝 #
猜你喜歡