win10+GTX1050+pytorch安装
泪目了,家人们!!!
我终于搞定了pytorch的GPU版本,
太难了,我搞了很久
CUDA_torch110_tf190是我的环境名字,意在醒目的提醒自己这是有CUDA的环境,而且torch版本是1.10版本,tensorflow版本是1.90版本,但是tensorflow貌似没成功装上,算了
感觉现在还是pytorch用的比较多
下面是一些常用的检测有CUDA版本的pytorch是否成功安装的命令。
import torch
torch.cuda.is_available()
torch.cuda.current_device()
torch.cuda.device(0)
torch.cuda.device_count()
点个赞呗,祝大家都能又快又好的弄好环境!