win10+Anaconda3安裝pytorch/torchvision cpu/gpu版本

我在運行一個代碼的時候出錯提示需要導入torvision,所以就進行了一些百度。因爲我之前安裝的tensorflow是gpu版本的,然後發現百度上大部分都是cpu版本的安裝pytorch方法。最後找到了一種方法可以安裝gpu版本的pytorch。首先你需要在nvidia控制面板裏查看你的cuda支持哪個版本,然後到https://pytorch.org/進行查找。 因爲我的cuda版本是10.1,
相關文章
相關標籤/搜索