Skip to content
This repository was archived by the owner on Dec 28, 2023. It is now read-only.

Commit ccc4177

Browse files
author
Omegastick
committed
Update for latest PyTorch version
1 parent 9eda405 commit ccc4177

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

example/gym_client.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
#include <spdlog/spdlog.h>
66
#include <spdlog/sinks/basic_file_sink.h>
7+
#include <ATen/Parallel.h>
78

89
#include <cpprl/cpprl.h>
910

@@ -67,7 +68,7 @@ int main(int argc, char *argv[])
6768
spdlog::set_level(spdlog::level::debug);
6869
spdlog::set_pattern("%^[%T %7l] %v%$");
6970

70-
torch::set_num_threads(1);
71+
at::set_num_threads(1);
7172
torch::manual_seed(0);
7273

7374
torch::Device device = use_cuda ? torch::kCUDA : torch::kCPU;

0 commit comments

Comments
 (0)