Answers for "pytorch version 1.7.1 cuda"

1

get pytorch version

import torch
print(torch.__version__)
Posted by: Guest on April-26-2020
2

cuda 10 install pytorch

# CUDA 9.2
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch
Posted by: Guest on May-16-2020

Python Answers by Framework

Browse Popular Code Answers by Language