1.首先查看自己下载的pytorch版本
jupyter notebook 下输入,就可以查看自己下载的版本了
import torch
print(torch.__version__)
2.如果是CPU版本的话们需要更改成GPU版本
去pytorch官网,然后进行如下选择
然后下载到自己的环境中去,下载完成后再次检查一下。
import torch
print(torch.__version__)
torch.cuda.is_available()
jupyter notebook 下输入,就可以查看自己下载的版本了
import torch
print(torch.__version__)
去pytorch官网,然后进行如下选择
然后下载到自己的环境中去,下载完成后再次检查一下。
import torch
print(torch.__version__)
torch.cuda.is_available()