PyTorch

CUDA 11.3. ROCm 4.2 (beta). CPU. Run this Command: conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch. Previous versions of PyTorch ... GetStarted Ecosystem Tools LearnaboutthetoolsandframeworksinthePyTorchEcosystem EcosystemDay-2021 Seetheposterspresentedatecosystemday2021 DeveloperDay-2021 Seetheposterspresentedatdeveloperday2021 Mobile Blog Tutorials Docs PyTorch torchaudio torchtext torchvision TorchData TorchRec TorchServe PyTorchonXLADevices Resources About LearnaboutPyTorch’sfeaturesandcapabilities Community JointhePyTorchdevelopercommunitytocontribute,learn,andgetyourquestionsanswered. Events Findevents,webinars,andpodcasts DeveloperResources Findresourcesandgetquestionsanswered Forums AplacetodiscussPyTorchcode,issues,install,research Models(Beta) Discover,publish,andreusepre-trainedmodels GitHub X From       ResearchToProduction Anopensourcemachinelearningframeworkthatacceleratesthepathfromresearchprototypingtoproductiondeployment. Install NEW!PyTorch1.11,TorchData,andfunctorcharenowavailable NEW!IntroducingTorchRec,andotherdomainlibraryupdatesinPyTorch1.11 NEW!IntroducingTorchRec,alibraryformodernproductionrecommendationsystems KeyFeatures&Capabilities SeeallFeatures ProductionReady TransitionseamlesslybetweeneagerandgraphmodeswithTorchScript,andacceleratethepathtoproductionwithTorchServe. DistributedTraining Scalabledistributedtrainingandperformanceoptimizationinresearchandproductionisenabledbythetorch.distributedbackend. RobustEcosystem ArichecosystemoftoolsandlibrariesextendsPyTorchandsupportsdevelopmentincomputervision,NLPandmore. CloudSupport PyTorchiswellsupportedonmajorcloudplatforms,providingfrictionlessdevelopmentandeasyscaling. SupportUkraine🇺🇦HelpProvideHumanitarianAidtoUkraine. InstallPyTorch Selectyourpreferencesandruntheinstallcommand.StablerepresentsthemostcurrentlytestedandsupportedversionofPyTorch.Thisshould besuitableformanyusers.Previewisavailableifyouwantthelatest,notfullytestedandsupported,1.11buildsthataregeneratednightly. Pleaseensurethatyouhavemettheprerequisitesbelow(e.g.,numpy),dependingonyourpackagemanager.Anacondaisourrecommended packagemanagersinceitinstallsalldependencies.Youcanalso installpreviousversionsofPyTorch.NotethatLibTorchisonlyavailableforC++. AdditionalsupportorwarrantyforsomePyTorchStableandLTSbinariesareavailablethroughthePyTorchEnterpriseSupportProgram. PyTorchBuild YourOS Package Language ComputePlatform RunthisCommand: PyTorchBuild Stable(1.11.0) Preview(Nightly) LTS(1.8.2) YourOS Linux Mac Windows Package Conda Pip LibTorch Source Language Python C++/Java ComputePlatform CUDA10.2 CUDA11.3 ROCm4.2(beta) CPU RunthisCommand: condainstallpytorchtorchvision-cpytorch PreviousversionsofPyTorch QuickStartWithCloudPartners GetupandrunningwithPyTorchquicklythroughpopularcloudplatformsandmachinelearningservices. AlibabaCloud AlibabaCloudPAI AmazonWebServices PyTorchonAWS AmazonSageMaker AWSDeepLearningContainers AWSDeepLearningAMIs GoogleCloudPlatform CloudDeepLearningVMImage DeepLearningContainers MicrosoftAzure- PyTorchEnterpriseProgram PyTorchonAzure AzureMachineLearning AzureFunctions Ecosystem FeatureProjects SeeallProjects Explorearichecosystemoflibraries,tools,andmoretosupportdevelopment. Captum Captum(“comprehension”inLatin)isanopensource,extensiblelibraryformodelinterpretabilitybuiltonPyTorch. PyTorchGeometric PyTorchGeometricisalibraryfordeeplearningonirregularinputdatasuchasgraphs,pointclouds,andmanifolds. skorch skorchisahigh-levellibraryforPyTorchthatprovidesfullscikit-learncompatibility. Community JointhePyTorchdevelopercommunitytocontribute,learn,andgetyourquestionsanswered. PyTorchDiscuss BrowseandjoindiscussionsondeeplearningwithPyTorch. Slack Discussadvancedtopics.Requestaccess:https://bit.ly/ptslack 中文文档 DocsandtutorialsinChinese,translatedbythecommunity. 파이토치(PyTorch) TutorialsinKorean,translatedbythecommunity. 日本語(PyTorch) TutorialsinJapanese,translatedbythecommunity. Contributors StayuptodatewiththecodebaseanddiscoverRFCs,PRsandmore Companies&UniversitiesUsingPyTorch Reduceinferencecostsby71%anddrivescaleoutusingPyTorch,TorchServe,andAWSInferentia. LearnMore PushingthestateoftheartinNLPandMulti-tasklearning. LearnMore UsingPyTorch’sflexibilitytoefficientlyresearchnewalgorithmicapproaches. LearnMore FollowUsonTwitter LoadingTwittertimeline... Docs AccesscomprehensivedeveloperdocumentationforPyTorch ViewDocs Tutorials Getin-depthtutorialsforbeginnersandadvanceddevelopers ViewTutorials Resources Finddevelopmentresourcesandgetyourquestionsanswered ViewResources GetStarted Ecosystem Tools EcosystemDay2021 DeveloperDay2021 Mobile Blog Tutorials Docs PyTorch torchaudio torchtext torchvision TorchElastic TorchServe PyTorchonXLADevices Resources About Community Events DeveloperResources Forum Models(Beta) GitHub Toanalyzetrafficandoptimizeyourexperience,weservecookiesonthissite.Byclickingornavigating,youagreetoallowourusageofcookies.Asthecurrentmaintainersofthissite,Facebook’sCookiesPolicyapplies.Learnmore,includingaboutavailablecontrols:CookiesPolicy.