linux下安裝指定版本的gpu版本的pytorch與torchvison

記住了 就這條命令,以pytorch0.4.1,cuda10,torchvison0.2.1爲例,可以自由更換版本

conda install pytorch=0.4.1 torchvision=0.2.1 cudatoolkit=10.0 -c pytorch

 

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